Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dscatter_test.cc30 Literal* scatter_indices, Literal* updates) { in RunTest() argument
31 RunTest(hlo_text, {operand, scatter_indices, updates}); in RunTest()
66 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
68 RunTest(hlo_text, &operand, &scatter_indices, &updates); in XLA_TEST_F()
97 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 1}); in XLA_TEST_F() local
99 RunTest(hlo_text, &operand, &scatter_indices, &updates); in XLA_TEST_F()
125 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
128 RunTest(hlo_text, &operand, &scatter_indices, &updates); in XLA_TEST_F()
194 Literal scatter_indices = LiteralUtil::CreateR2<int32>({{0, 0, 0}}); in XLA_TEST_F() local
195 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.cc3434 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
3437 Evaluate({&operand, &scatter_indices, &updates})); in TEST_F()
3467 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
3471 Evaluate({&operand, &scatter_indices, &updates})); in TEST_F()
3502 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
3505 Evaluate({&operand, &scatter_indices, &updates})); in TEST_F()
3536 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
3539 Evaluate({&operand, &scatter_indices, &updates})); in TEST_F()
3570 Literal scatter_indices = LiteralUtil::CreateR1<int32>({2, 1}); in TEST_P() local
3574 Evaluate({&operand, &scatter_indices, &updates})); in TEST_P()
[all …]
Ddynamic_padder_test.cc272 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
278 {&operand, &scatter_indices, &updates, &dynamic_size}); in XLA_TEST_F()
334 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
352 {&operand, &scatter_indices, &updates, &dynamic_size}); in XLA_TEST_F()
Dhlo_parser_test.cc877 ENTRY %Scatter (input_tensor: f32[50,49,48,47,46], scatter_indices: s64[10,9,8,7,5], updates: f32[1…
879 %scatter_indices = s64[10,9,8,7,5]{4,3,2,1,0} parameter(1)
881 …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…
896 ENTRY %Scatter (input_tensor: f32[50,49,48,47,46], scatter_indices: s64[10,9,8,7,5], updates: f32[1…
898 %scatter_indices = s64[10,9,8,7,5]{4,3,2,1,0} parameter(1)
900 …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…
915 ENTRY %Scatter (input_tensor: f32[50,49,48,47,46], scatter_indices: s64[10,9,8,7,5], updates: f32[1…
917 %scatter_indices = s64[10,9,8,7,5]{4,3,2,1,0} parameter(1)
919 …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…
Dhlo_evaluator_typed_visitor.h1964 const Shape& updates_shape, const Literal* scatter_indices)
1965 : dim_numbers_(*dim_numbers), scatter_indices_(*scatter_indices) {
2170 TF_ASSIGN_OR_RETURN(const Literal& scatter_indices,
2192 updates_shape, &scatter_indices);
Dhlo_instruction_test.cc1507 HloInstruction* scatter_indices = in TEST_F() local
1526 input_tensor_shape, input, scatter_indices, scatter_updates, in TEST_F()
Dhlo_instruction.h880 HloInstruction* scatter_indices, HloInstruction* updates,
Dhlo_instructions.cc2534 HloInstruction* scatter_indices, HloInstruction* updates, in HloScatterInstruction() argument
2542 AppendOperand(scatter_indices); in HloScatterInstruction()
Dhlo_instructions.h1496 HloInstruction* scatter_indices, HloInstruction* updates,
Dhlo_instruction.cc1428 HloInstruction* scatter_indices, HloInstruction* updates, in CreateScatter() argument
1433 shape, operand, scatter_indices, updates, update_computation, in CreateScatter()
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py435 def scatter(operand, scatter_indices, updates, update_computation, argument
439 scatter_indices,
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client.py1621 scatter_indices, argument
1628 return ops.Scatter(a, scatter_indices, updates,
Dxla_client_test.py1965 scatter_indices = np.array([0, 2], dtype=np.int32)
1976 c.Constant(a), c.Constant(scatter_indices), c.Constant(updates),
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md2277 array `operand`, with several slices (at indices specified by `scatter_indices`)
2283 <b> `scatter(operand, scatter_indices, updates, update_computation, index_vector_dim, update_window…
2288 `scatter_indices` | `XlaOp` | Array containing the starting indices of the…
2291 `index_vector_dim` | `int64` | The dimension in `scatter_indices` that cont…
2297 If `index_vector_dim` is equal to `scatter_indices.rank` we implicitly consider
2298 `scatter_indices` to have a trailing `1` dimension.
2307 scatter_indices.rank - 1`.
2321 `scatter_indices`, skipping `index_vector_dim` (i.e.
2322 `scatter_indices.shape.dims`[`k`], if `k` < `index_vector_dim` and
2323 `scatter_indices.shape.dims`[`k+1`] otherwise).
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc845 const HloInstruction* scatter_indices = scatter->operand(1); in HandleScatter() local
868 return GetIrArray(*scatter_indices, *scatter) in HandleScatter()
893 const HloInstruction* scatter_indices = scatter->operand(1); in EmitScatter() local
937 Shape scatter_indices_shape = scatter_indices->shape(); in EmitScatter()
965 scatter_indices_shape, scatter_indices->shape(), &b_))); in EmitScatter()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h587 XlaOp Scatter(XlaOp input, XlaOp scatter_indices, XlaOp updates,
1008 friend XlaOp Scatter(XlaOp input, XlaOp scatter_indices, XlaOp updates,
1888 XlaOp Scatter(XlaOp input, XlaOp scatter_indices, XlaOp updates,
Dxla_builder.cc1802 XlaOp XlaBuilder::Scatter(XlaOp input, XlaOp scatter_indices, XlaOp updates, in Scatter() argument
1814 GetShapePtr(scatter_indices)); in Scatter()
1828 {input, scatter_indices, updates}); in Scatter()
3497 XlaOp Scatter(const XlaOp input, const XlaOp scatter_indices, in Scatter() argument
3501 return input.builder()->Scatter(input, scatter_indices, updates, in Scatter()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dexport.mlir668 func @main(%input_tensor: tensor<200x100x300xf32>, %scatter_indices: tensor<10x2xi32>, %updates: te…
669 %0 = "xla_hlo.scatter" (%input_tensor, %scatter_indices, %updates) ({
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc704 value_map[op.operand()], value_map[op.scatter_indices()], in ExportXlaOp()
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dhlo_ops.td1018 HLO_Tensor:$scatter_indices,
Dhlo_ops_base.td978 with several slices (at indices specified by `scatter_indices`)