/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | scatter_test.cc | 30 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/ |
D | scatter_expander.cc | 32 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 …]
|
D | hlo_evaluator_test.cc | 3538 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 …]
|
D | dynamic_padder_test.cc | 428 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()
|
D | hlo_parser_test.cc | 941 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()
|
D | hlo_evaluator_typed_visitor.h | 2130 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);
|
D | hlo_instruction_test.cc | 1565 HloInstruction* scatter_indices = in TEST_F() local 1584 input_tensor_shape, input, scatter_indices, scatter_updates, in TEST_F()
|
D | hlo_instruction.h | 951 HloInstruction* scatter_indices, HloInstruction* updates,
|
D | hlo_instructions.cc | 2818 HloInstruction* scatter_indices, HloInstruction* updates, in HloScatterInstruction() argument 2826 AppendOperand(scatter_indices); in HloScatterInstruction()
|
D | hlo_instructions.h | 1702 HloInstruction* scatter_indices, HloInstruction* updates,
|
D | hlo_instruction.cc | 1626 HloInstruction* scatter_indices, HloInstruction* updates, in CreateScatter() argument 1631 shape, operand, scatter_indices, updates, update_computation, in CreateScatter()
|
/external/tensorflow/tensorflow/compiler/tf2xla/python/ |
D | xla.py | 493 def scatter(operand, scatter_indices, updates, update_computation, argument 497 scatter_indices,
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
D | mlir_hlo_builder.h | 185 const Shape& shape, XlaOp input, XlaOp scatter_indices, XlaOp updates,
|
D | mlir_hlo_builder.cc | 301 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/ |
D | mlir_hlo_to_hlo.cc | 949 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/ |
D | operation_semantics.md | 2387 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/ |
D | xla_builder.h | 856 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,
|
D | xla_builder.cc | 2302 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/ |
D | lhlo_ops.td | 469 Arg<LHLO_Buffer, "", [MemRead]>:$scatter_indices,
|
D | hlo_ops.td | 1090 HLO_Tensor:$scatter_indices,
|
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
D | import.hlotxt | 766 %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] …
|
D | export.mlir | 757 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/ |
D | legalize_hlo.cc | 1095 Value indices = scatter_op.scatter_indices(); in matchAndRewrite()
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | xla_client_test.py | 1833 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/ |
D | ir_emitter_unnested.cc | 2531 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()
|