| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | scatter_test.cc | 31 Literal* scatter_indices, Literal* updates) { in RunTest()
67 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
98 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 1}); in XLA_TEST_F() local
126 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
195 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 0, 0}}); in XLA_TEST_F() local
223 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
252 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
281 Literal scatter_indices = LiteralUtil::CreateR1<int32>({2, 1}); in XLA_TEST_F() local
311 Literal scatter_indices = LiteralUtil::CreateR1<int32>({1, 1}); in XLA_TEST_F() local
340 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 2}, {2, 1}}); in XLA_TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander.cc | 32 HloInstruction* scatter_indices, int64 index_vector_dim) { in TransposeIndexVectorDimToLast()
57 HloInstruction* scatter_indices, int64 index_vector_dim) { in CanonicalizeScatterIndices()
226 HloInstruction* scatter_indices = loop_state[1]; in ScatterLoopBody() local
348 HloInstruction* scatter_indices = scatter->mutable_operand(1); in ExpandScatter() local
|
| D | hlo_evaluator_test.cc | 2335 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
2368 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
2403 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
2437 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
2471 Literal scatter_indices = LiteralUtil::CreateR1<int32>({2, 1}); in TEST_P() local
2507 Literal scatter_indices = LiteralUtil::CreateR1<int32>({1, 1}); in TEST_F() local
2541 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 2}, {2, 1}}); in TEST_F() local
2577 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}}); in TEST_F() local
2615 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}}); in TEST_F() local
2650 Literal scatter_indices = LiteralUtil::CreateR1<int32>({1, 1}); in TEST_F() local
[all …]
|
| D | hlo_evaluator_typed_visitor.h | 2107 const Shape& updates_shape, const Literal* scatter_indices) in UpdateScatterIndexToInputIndex()
|
| D | hlo_instruction_test.cc | 1434 HloInstruction* scatter_indices = in TEST_F() local
|
| D | hlo_instruction.cc | 1316 HloInstruction* scatter_indices, HloInstruction* updates, in CreateScatter()
|
| D | hlo_instructions.cc | 2396 HloInstruction* scatter_indices, HloInstruction* updates, in HloScatterInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 763 const LocalOp& input, const LocalOp& scatter_indices, in Scatter()
|
| D | xla_client.py | 1779 def Scatter(self, a, scatter_indices, updates, update_computation, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1031 const HloInstruction* scatter_indices = scatter->operand(1); in HandleScatter() local
1079 const HloInstruction* scatter_indices = scatter->operand(1); in EmitScatter() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1864 XlaOp XlaBuilder::Scatter(const XlaOp& input, const XlaOp& scatter_indices, in Scatter()
3431 XlaOp Scatter(const XlaOp& input, const XlaOp& scatter_indices, in Scatter()
|