Searched refs:TorchGather (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing_test.cc | 115 TorchGather(input, index, 1); in XLA_TEST_F()
129 TorchGather(input, index, 1, false); in XLA_TEST_F()
|
| D | slicing.h | 65 XlaOp TorchGather(XlaOp input, XlaOp index, int64 dim, bool sparse = true);
|
| D | slicing.cc | 158 XlaOp TorchGather(XlaOp input, XlaOp index, int64 dim, bool sparse) { in TorchGather() function
|