Commit 34883b6f authored by Eugene Brevdo's avatar Eugene Brevdo Committed by TensorFlower Gardener
Browse files

Extend gather_nd to support sliced indexing. 

Includes:
  * Some bugfixes for empty inputs
  * Better parallelization on GPU using a CUDA kernel.
Change: 128236016
parent 59a796d9
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment