Home
last modified time | relevance | path

Searched defs:scatter_indices (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dscatter_test.cc28 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/
Dscatter_expander.cc35 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
Dhlo_evaluator_test.cc3576 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 …]
Ddynamic_padder_test.cc585 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
Dhlo_instructions.h2174 const HloInstruction* scatter_indices() const { in scatter_indices() function
2177 HloInstruction* scatter_indices() { in scatter_indices() function
Dhlo_instruction.cc1937 HloInstruction* scatter_indices, HloInstruction* updates, in CreateScatter()
1948 HloInstruction* scatter_indices, absl::Span<HloInstruction* const> updates, in CreateScatter()
Dhlo_instruction_test.cc1664 HloInstruction* scatter_indices = in TEST_F() local
Dhlo_evaluator.cc2638 int64_t updates_rank, const Literal* scatter_indices) in UpdateScatterIndexToInputIndex()
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py582 def scatter(operand, scatter_indices, updates, update_computation, argument
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc395 const Shape& shape, absl::Span<const XlaOp> inputs, XlaOp scatter_indices, in ScatterInternal()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2427 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/
Dmlir_hlo_to_hlo.cc1486 xla::XlaOp scatter_indices; in ExportXlaOp() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2245 const IrArray& scatter_indices = ir_arrays[0]; in EmitScatter() local