/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | scatter_test.cc | 28 Literal* scatter_indices, Literal* updates) { in RunTest() 64 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({0, 2}); in XLA_TEST_F() local 96 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({0, 1}); in XLA_TEST_F() local 125 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({0, 2}); in XLA_TEST_F() local 194 Literal scatter_indices = LiteralUtil::CreateR2<int32_t>({{0, 0, 0}}); in XLA_TEST_F() local 222 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({0, 2}); in XLA_TEST_F() local 253 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({0, 2}); in XLA_TEST_F() local 283 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({0, 2}); in XLA_TEST_F() local 313 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({2, 1}); in XLA_TEST_F() local 343 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({1, 1}); in XLA_TEST_F() local [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | scatter_expander.cc | 35 HloInstruction* scatter_indices, int64_t index_vector_dim) { in TransposeIndexVectorDimToLast() 60 HloInstruction* scatter_indices, int64_t index_vector_dim) { in CanonicalizeScatterIndices() 255 HloInstruction* scatter_indices = loop_state[operands.size()]; in ScatterLoopBody() local 382 const HloInstruction* scatter_indices = scatter->scatter_indices(); in ScatterTripCount() local 416 HloInstruction* scatter_indices = scatter->scatter_indices(); in ExpandInstruction() local
|
D | hlo_evaluator_test.cc | 3576 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({0, 2}); in TEST_F() local 3610 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({0, 2}); in TEST_F() local 3645 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({0, 2}); in TEST_F() local 3680 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({0, 2}); in TEST_F() local 3716 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({2, 1}); in TEST_P() local 3752 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({1, 1}); in TEST_F() local 3787 Literal scatter_indices = LiteralUtil::CreateR2<int32_t>({{0, 2}, {2, 1}}); in TEST_F() local 3824 Literal scatter_indices = LiteralUtil::CreateR2<int32_t>({{0, 0}, {1, 0}}); in TEST_F() local 3862 Literal scatter_indices = LiteralUtil::CreateR2<int32_t>({{0, 0}, {1, 0}}); in TEST_F() local 3897 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({1, 1}); in TEST_F() local [all …]
|
D | dynamic_padder_test.cc | 585 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({0, 2}); in XLA_TEST_F() local 648 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({0}); in XLA_TEST_F() local 689 Literal scatter_indices = LiteralUtil::CreateR1<int32_t>({0, 2}); in XLA_TEST_F() local
|
D | hlo_instructions.h | 2174 const HloInstruction* scatter_indices() const { in scatter_indices() function 2177 HloInstruction* scatter_indices() { in scatter_indices() function
|
D | hlo_instruction.cc | 1937 HloInstruction* scatter_indices, HloInstruction* updates, in CreateScatter() 1948 HloInstruction* scatter_indices, absl::Span<HloInstruction* const> updates, in CreateScatter()
|
D | hlo_instruction_test.cc | 1664 HloInstruction* scatter_indices = in TEST_F() local
|
D | hlo_evaluator.cc | 2638 int64_t updates_rank, const Literal* scatter_indices) in UpdateScatterIndexToInputIndex()
|
/external/tensorflow/tensorflow/compiler/tf2xla/python/ |
D | xla.py | 582 def scatter(operand, scatter_indices, updates, update_computation, argument
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
D | mlir_hlo_builder.cc | 395 const Shape& shape, absl::Span<const XlaOp> inputs, XlaOp scatter_indices, in ScatterInternal()
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.cc | 2427 XlaOp XlaBuilder::Scatter(XlaOp input, XlaOp scatter_indices, XlaOp updates, in Scatter() 2436 XlaOp XlaBuilder::Scatter(absl::Span<const XlaOp> inputs, XlaOp scatter_indices, in Scatter() 2475 const Shape& shape, absl::Span<const XlaOp> inputs, XlaOp scatter_indices, in ScatterInternal() 4906 XlaOp Scatter(const XlaOp input, const XlaOp scatter_indices, in Scatter() 4915 XlaOp Scatter(absl::Span<const XlaOp> inputs, XlaOp scatter_indices, in Scatter()
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ |
D | mlir_hlo_to_hlo.cc | 1486 xla::XlaOp scatter_indices; in ExportXlaOp() local
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emitter_unnested.cc | 2245 const IrArray& scatter_indices = ir_arrays[0]; in EmitScatter() local
|