Compilation of loss function and reconstruction in tf.while_loop has been re-written in terms of tf.where (as opposed to multiplication by 0/1 not_finished values).
Compilation of loss function and reconstruction in tf.while_loop has been re-written in terms of tf.where (as opposed to multiplication by 0/1 not_finished values).

aakhundov/tf-attend-infer-repeat