Commit 47694b7e authored by Blake Hechtman's avatar Blake Hechtman Committed by TensorFlower Gardener
Browse files

[XLA:CLIENT] Implement a varient of gather equivalent to torch.gather 

(https://pytorch.org/docs/stable/torch.html#torch.gather).

PiperOrigin-RevId: 234056021
parent 2caa84e5
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