Home
last modified time | relevance | path

Searched refs:scatter_indices (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dscatter_test.cc31 Literal* scatter_indices, Literal* updates) { in RunTest() argument
32 RunTest(hlo_text, {operand, scatter_indices, updates}); in RunTest()
67 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
69 RunTest(hlo_text, &operand, &scatter_indices, &updates); in XLA_TEST_F()
98 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 1}); in XLA_TEST_F() local
100 RunTest(hlo_text, &operand, &scatter_indices, &updates); in XLA_TEST_F()
126 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
129 RunTest(hlo_text, &operand, &scatter_indices, &updates); in XLA_TEST_F()
195 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 0, 0}}); in XLA_TEST_F() local
196 RunTest(hlo_text, &operand, &scatter_indices, &updates); in XLA_TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dscatter_expander.cc32 HloInstruction* scatter_indices, int64 index_vector_dim) { in TransposeIndexVectorDimToLast() argument
33 const Shape& scatter_indices_shape = scatter_indices->shape(); in TransposeIndexVectorDimToLast()
36 return scatter_indices; in TransposeIndexVectorDimToLast()
40 return scatter_indices; in TransposeIndexVectorDimToLast()
51 return MakeTransposeHlo(scatter_indices, permutation); in TransposeIndexVectorDimToLast()
57 HloInstruction* scatter_indices, int64 index_vector_dim) { in CanonicalizeScatterIndices() argument
61 TransposeIndexVectorDimToLast(scatter_indices, index_vector_dim)); in CanonicalizeScatterIndices()
62 if (scatter_indices->shape().rank() == index_vector_dim + 1 && in CanonicalizeScatterIndices()
63 scatter_indices->shape().dimensions(index_vector_dim) == 1) { in CanonicalizeScatterIndices()
65 ShapeUtil::DeleteDimension(index_vector_dim, scatter_indices->shape()); in CanonicalizeScatterIndices()
[all …]
Dhlo_evaluator_test.cc2335 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
2338 Evaluate({&operand, &scatter_indices, &updates})); in TEST_F()
2368 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
2372 Evaluate({&operand, &scatter_indices, &updates})); in TEST_F()
2403 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
2406 Evaluate({&operand, &scatter_indices, &updates})); in TEST_F()
2437 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
2440 Evaluate({&operand, &scatter_indices, &updates})); in TEST_F()
2471 Literal scatter_indices = LiteralUtil::CreateR1<int32>({2, 1}); in TEST_P() local
2475 Evaluate({&operand, &scatter_indices, &updates})); in TEST_P()
[all …]
Dhlo_parser_test.cc868 ENTRY %Scatter (input_tensor: f32[50,49,48,47,46], scatter_indices: s64[10,9,8,7,5], updates: f32[1… in CreateTestCases()
870 %scatter_indices = s64[10,9,8,7,5]{4,3,2,1,0} parameter(1) in CreateTestCases()
872 …0,49,48,47,46]{4,3,2,1,0} %input_tensor, s64[10,9,8,7,5]{4,3,2,1,0} %scatter_indices, f32[10,9,8,7… in CreateTestCases()
Dhlo_evaluator_typed_visitor.h2107 const Shape& updates_shape, const Literal* scatter_indices) in UpdateScatterIndexToInputIndex() argument
2108 : dim_numbers_(*dim_numbers), scatter_indices_(*scatter_indices) { in UpdateScatterIndexToInputIndex()
2313 TF_ASSIGN_OR_RETURN(const Literal& scatter_indices, in HandleScatter()
2335 updates_shape, &scatter_indices); in HandleScatter()
Dhlo_instruction_test.cc1434 HloInstruction* scatter_indices = in TEST_F() local
1453 input_tensor_shape, input, scatter_indices, scatter_updates, in TEST_F()
Dhlo_instruction.h733 HloInstruction* scatter_indices, HloInstruction* updates,
Dhlo_instructions.cc2396 HloInstruction* scatter_indices, HloInstruction* updates, in HloScatterInstruction() argument
2401 AppendOperand(scatter_indices); in HloScatterInstruction()
Dhlo_instructions.h1403 HloInstruction* scatter_indices, HloInstruction* updates,
Dhlo_instruction.cc1316 HloInstruction* scatter_indices, HloInstruction* updates, in CreateScatter() argument
1320 shape, operand, scatter_indices, updates, update_computation, in CreateScatter()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md2109 array `operand`, with several slices (at indices specified by `scatter_indices`)
2115 <b> `scatter(operand, scatter_indices, updates, update_computation, index_vector_dim, update_window…
2120 `scatter_indices` | `XlaOp` | Array containing the starting indices of the…
2123 `index_vector_dim` | `int64` | The dimension in `scatter_indices` that cont…
2128 If `index_vector_dim` is equal to `scatter_indices.rank` we implicitly consider
2129 `scatter_indices` to have a trailing `1` dimension.
2138 scatter_indices.rank - 1`.
2152 `scatter_indices`, skipping `index_vector_dim` (i.e.
2153 `scatter_indices.shape.dims`[`k`], if `k` < `index_vector_dim` and
2154 `scatter_indices.shape.dims`[`k+1`] otherwise).
[all …]
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc763 const LocalOp& input, const LocalOp& scatter_indices, in Scatter() argument
766 return xla::Scatter(input.op(), scatter_indices.op(), updates.op(), in Scatter()
Dlocal_computation_builder.h377 LocalOp Scatter(const LocalOp& input, const LocalOp& scatter_indices,
Dxla_client.py1779 def Scatter(self, a, scatter_indices, updates, update_computation, argument
1783 a, scatter_indices, updates, update_computation.computation,
Dxla_client_test.py1680 scatter_indices = np.array([0, 2], dtype=np.int32)
1690 c.Scatter(c.Constant(a), c.Constant(scatter_indices), c.Constant(updates),
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h547 XlaOp Scatter(const XlaOp& input, const XlaOp& scatter_indices,
971 friend XlaOp Scatter(const XlaOp& input, const XlaOp& scatter_indices,
1820 XlaOp Scatter(const XlaOp& input, const XlaOp& scatter_indices,
Dxla_builder.cc1864 XlaOp XlaBuilder::Scatter(const XlaOp& input, const XlaOp& scatter_indices, in Scatter() argument
1873 GetShape(scatter_indices)); in Scatter()
1887 {input, scatter_indices, updates}); in Scatter()
3431 XlaOp Scatter(const XlaOp& input, const XlaOp& scatter_indices, in Scatter() argument
3434 return input.builder()->Scatter(input, scatter_indices, updates, in Scatter()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1031 const HloInstruction* scatter_indices = scatter->operand(1); in HandleScatter() local
1054 return GetIrArray(*scatter_indices, *scatter) in HandleScatter()
1079 const HloInstruction* scatter_indices = scatter->operand(1); in EmitScatter() local
1123 Shape scatter_indices_shape = scatter_indices->shape(); in EmitScatter()
1151 scatter_indices_shape, scatter_indices->shape(), &b_))); in EmitScatter()