Fix gather for nonempty indices, empty slices
tf.gather([0], [[]]) now works. This required 1. Avoiding an empty GPU kernel launch. 2. Avoiding the use of -1 reshapes in the gradient. This code was also simplified a bit using the fact that array_ops.shape does it's own checking for fully defined shapes. Change: 130554783
Loading
Please sign in to comment