Home
last modified time | relevance | path

Searched refs:TorchGather (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dslicing_test.cc105 XLA_TEST_F(SlicingTest, TorchGather) { in XLA_TEST_F() argument
113 TorchGather(input, index, 1); in XLA_TEST_F()
Dslicing.h58 XlaOp TorchGather(XlaOp input, XlaOp index, int64 dim);
Dslicing.cc137 XlaOp TorchGather(XlaOp input, XlaOp index, int64 dim) { in TorchGather() function