| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | scatter_test.cc | 30 Literal* scatter_indices, Literal* updates) { in RunTest()
66 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
97 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 1}); in XLA_TEST_F() local
125 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
194 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 0, 0}}); in XLA_TEST_F() local
222 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>({0, 2}); in XLA_TEST_F() local
310 Literal scatter_indices = LiteralUtil::CreateR1<int32>({2, 1}); in XLA_TEST_F() local
340 Literal scatter_indices = LiteralUtil::CreateR1<int32>({1, 1}); in XLA_TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander.cc | 32 HloInstruction* scatter_indices, int64_t index_vector_dim) { in TransposeIndexVectorDimToLast()
57 HloInstruction* scatter_indices, int64_t index_vector_dim) { in CanonicalizeScatterIndices()
226 HloInstruction* scatter_indices = loop_state[1]; in ScatterLoopBody() local
331 HloInstruction* scatter_indices = scatter->mutable_operand(1); in ScatterTripCount() local
364 HloInstruction* scatter_indices = scatter->mutable_operand(1); in ExpandInstruction() local
|
| D | hlo_evaluator_test.cc | 3538 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
3571 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
3606 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
3640 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
3674 Literal scatter_indices = LiteralUtil::CreateR1<int32>({2, 1}); in TEST_P() local
3710 Literal scatter_indices = LiteralUtil::CreateR1<int32>({1, 1}); in TEST_F() local
3744 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 2}, {2, 1}}); in TEST_F() local
3780 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}}); in TEST_F() local
3818 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 0}, {1, 0}}); in TEST_F() local
3853 Literal scatter_indices = LiteralUtil::CreateR1<int32>({1, 1}); in TEST_F() local
[all …]
|
| D | dynamic_padder_test.cc | 501 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
563 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0}); in XLA_TEST_F() local
604 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
|
| D | hlo_instruction_test.cc | 1565 HloInstruction* scatter_indices = in TEST_F() local
|
| D | hlo_instruction.cc | 1813 HloInstruction* scatter_indices, HloInstruction* updates, in CreateScatter()
|
| D | hlo_instructions.cc | 2982 HloInstruction* scatter_indices, HloInstruction* updates, in HloScatterInstruction()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 561 def scatter(operand, scatter_indices, updates, update_computation, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 318 const Shape& shape, XlaOp input, XlaOp scatter_indices, XlaOp updates, in ScatterInternal()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 1068 xla::XlaOp operand, scatter_indices, updates; in ExportXlaOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2330 XlaOp XlaBuilder::Scatter(XlaOp input, XlaOp scatter_indices, XlaOp updates, in Scatter()
2352 const Shape& shape, XlaOp input, XlaOp scatter_indices, XlaOp updates, in ScatterInternal()
4744 XlaOp Scatter(const XlaOp input, const XlaOp scatter_indices, in Scatter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2385 const IrArray& scatter_indices = ir_arrays[0]; in EmitScatter() local
|