Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dslicing_test.cc130 TorchIndexSelect(input, index, 0); in XLA_TEST_F()
149 TorchIndexSelect(input, index, 1); in XLA_TEST_F()
Dslicing.h66 XlaOp TorchIndexSelect(XlaOp input, XlaOp index, int64 dim);
Dslicing.cc164 XlaOp TorchIndexSelect(XlaOp input, XlaOp index, int64 dim) { in TorchIndexSelect() function