Searched refs:TorchScatterDense (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | slicing_test.cc | 135 XLA_TEST_F(SlicingTest, TorchScatterDense) { in XLA_TEST_F() argument
145 TorchScatterDense(input, index, src, 1, in XLA_TEST_F()
|
| D | slicing.h | 71 XlaOp TorchScatterDense(XlaOp input, XlaOp index, XlaOp src, int64 dim,
|
| D | slicing.cc | 226 XlaOp TorchScatterDense(XlaOp input, XlaOp index, XlaOp src, int64 dim, in TorchScatterDense() function
|