Home
last modified time | relevance | path

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

12

/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.cc3538 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
3541 Evaluate({&operand, &scatter_indices, &updates})); in TEST_F()
3571 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
3575 Evaluate({&operand, &scatter_indices, &updates})); in TEST_F()
3606 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
3609 Evaluate({&operand, &scatter_indices, &updates})); in TEST_F()
3640 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in TEST_F() local
3643 Evaluate({&operand, &scatter_indices, &updates})); in TEST_F()
3674 Literal scatter_indices = LiteralUtil::CreateR1<int32>({2, 1}); in TEST_P() local
3678 Evaluate({&operand, &scatter_indices, &updates})); in TEST_P()
[all …]
Ddynamic_padder_test.cc428 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
434 {&operand, &scatter_indices, &updates, &dynamic_size}); in XLA_TEST_F()
490 Literal scatter_indices = LiteralUtil::CreateR1<int32>({0, 2}); in XLA_TEST_F() local
508 {&operand, &scatter_indices, &updates, &dynamic_size}); in XLA_TEST_F()
Dhlo_parser_test.cc941 ENTRY %Scatter (input_tensor: f32[50,49,48,47,46], scatter_indices: s64[10,9,8,7,5], updates: f32[1… in CreateTestCases()
943 %scatter_indices = s64[10,9,8,7,5]{4,3,2,1,0} parameter(1) in CreateTestCases()
945 …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()
960 ENTRY %Scatter (input_tensor: f32[50,49,48,47,46], scatter_indices: s64[10,9,8,7,5], updates: f32[1… in CreateTestCases()
962 %scatter_indices = s64[10,9,8,7,5]{4,3,2,1,0} parameter(1) in CreateTestCases()
964 …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()
979 ENTRY %Scatter (input_tensor: f32[50,49,48,47,46], scatter_indices: s64[10,9,8,7,5], updates: f32[1… in CreateTestCases()
981 %scatter_indices = s64[10,9,8,7,5]{4,3,2,1,0} parameter(1) in CreateTestCases()
983 …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.h2130 const Shape& updates_shape, const Literal* scatter_indices)
2131 : dim_numbers_(*dim_numbers), scatter_indices_(*scatter_indices) {
2336 TF_ASSIGN_OR_RETURN(const Literal& scatter_indices,
2356 updates_shape, &scatter_indices);
Dhlo_instruction_test.cc1565 HloInstruction* scatter_indices = in TEST_F() local
1584 input_tensor_shape, input, scatter_indices, scatter_updates, in TEST_F()
Dhlo_instruction.h951 HloInstruction* scatter_indices, HloInstruction* updates,
Dhlo_instructions.cc2818 HloInstruction* scatter_indices, HloInstruction* updates, in HloScatterInstruction() argument
2826 AppendOperand(scatter_indices); in HloScatterInstruction()
Dhlo_instructions.h1702 HloInstruction* scatter_indices, HloInstruction* updates,
Dhlo_instruction.cc1626 HloInstruction* scatter_indices, HloInstruction* updates, in CreateScatter() argument
1631 shape, operand, scatter_indices, updates, update_computation, in CreateScatter()
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py493 def scatter(operand, scatter_indices, updates, update_computation, argument
497 scatter_indices,
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.h185 const Shape& shape, XlaOp input, XlaOp scatter_indices, XlaOp updates,
Dmlir_hlo_builder.cc301 const Shape& shape, XlaOp input, XlaOp scatter_indices, XlaOp updates, in ScatterInternal() argument
308 loc_, ty, GetValue(input), GetValue(scatter_indices), GetValue(updates), in ScatterInternal()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc949 xla::XlaOp operand, scatter_indices, updates; in ExportXlaOp() local
951 if (failed(GetXlaOp(op.scatter_indices(), value_map, &scatter_indices, op))) in ExportXlaOp()
955 value_map[op] = xla::Scatter(operand, scatter_indices, updates, in ExportXlaOp()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md2387 array `operand`, with several slices (at indices specified by `scatter_indices`)
2393 <b> `scatter(operand, scatter_indices, updates, update_computation, index_vector_dim, update_window…
2398 `scatter_indices` | `XlaOp` | Array containing the starting indices of the…
2401 `index_vector_dim` | `int64` | The dimension in `scatter_indices` that cont…
2407 If `index_vector_dim` is equal to `scatter_indices.rank` we implicitly consider
2408 `scatter_indices` to have a trailing `1` dimension.
2417 scatter_indices.rank - 1`.
2431 `scatter_indices`, skipping `index_vector_dim` (i.e.
2432 `scatter_indices.shape.dims`[`k`], if `k` < `index_vector_dim` and
2433 `scatter_indices.shape.dims`[`k+1`] otherwise).
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h856 XlaOp Scatter(XlaOp input, XlaOp scatter_indices, XlaOp updates,
862 const Shape& shape, XlaOp input, XlaOp scatter_indices, XlaOp updates,
1396 friend XlaOp Scatter(XlaOp input, XlaOp scatter_indices, XlaOp updates,
2447 XlaOp Scatter(XlaOp input, XlaOp scatter_indices, XlaOp updates,
Dxla_builder.cc2302 XlaOp XlaBuilder::Scatter(XlaOp input, XlaOp scatter_indices, XlaOp updates, in Scatter() argument
2309 GetShapePtr(scatter_indices)); in Scatter()
2317 return ScatterInternal(shape, input, scatter_indices, updates, in Scatter()
2324 const Shape& shape, XlaOp input, XlaOp scatter_indices, XlaOp updates, in ScatterInternal() argument
2337 {input, scatter_indices, updates}); in ScatterInternal()
4882 XlaOp Scatter(const XlaOp input, const XlaOp scatter_indices, in Scatter() argument
4886 return input.builder()->Scatter(input, scatter_indices, updates, in Scatter()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_ops.td469 Arg<LHLO_Buffer, "", [MemRead]>:$scatter_indices,
Dhlo_ops.td1090 HLO_Tensor:$scatter_indices,
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dimport.hlotxt766 %scatter_indices = s64[10,2] parameter(1)
768 … f32[200,100,300] scatter(f32[200,100,300] %input_tensor, s64[10,2] %scatter_indices, f32[10,300] …
Dexport.mlir757 func @main(%input_tensor: tensor<200x100x300xf32>, %scatter_indices: tensor<10x2xi32>, %updates: te…
758 %0 = "mhlo.scatter" (%input_tensor, %scatter_indices, %updates) ({
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc1095 Value indices = scatter_op.scatter_indices(); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla_client_test.py1833 scatter_indices = np.array([0, 2], dtype=np.int32)
1844 ops.Constant(c, a), ops.Constant(c, scatter_indices),
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2531 const IrArray& scatter_indices = ir_arrays[0]; in EmitScatterFromMlir() local
2543 return scatter_indices.EmitReadArrayElement(index, &b_, in EmitScatterFromMlir()
2582 desc.scatter_indices_shape = TypeToShape(scatter.scatter_indices().getType()); in EmitScatter()

12