/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
H A D | mhlo_flatten_tuple.mlir | 64 …%t0 = "mhlo.get_tuple_element"(%arg2) {index = 0 : i32} : (tuple<tensor<1xi32>, tensor<2xi32>, ten… 65 …%t1 = "mhlo.get_tuple_element"(%arg2) {index = 1 : i32} : (tuple<tensor<1xi32>, tensor<2xi32>, ten… 72 …%t0 = "mhlo.get_tuple_element"(%arg2) {index = 0 : i32} : (tuple<tensor<1xi32>, tensor<2xi32>, ten… 73 …%t1 = "mhlo.get_tuple_element"(%arg2) {index = 1 : i32} : (tuple<tensor<1xi32>, tensor<2xi32>, ten… 74 …%t2 = "mhlo.get_tuple_element"(%arg2) {index = 2 : i32} : (tuple<tensor<1xi32>, tensor<2xi32>, ten… 75 …%t3 = "mhlo.get_tuple_element"(%arg2) {index = 3 : i32} : (tuple<tensor<1xi32>, tensor<2xi32>, ten… 81 …%5 = "mhlo.get_tuple_element"(%1) {index = 3 : i32} : (tuple<tensor<1xi32>, tensor<2xi32>, tensor<… 111 …%t0_1 = "mhlo.get_tuple_element"(%arg1) {index = 0 : i32} : (tuple<tuple<tensor<1xi32>, tensor<2xi… 112 …%t0 = "mhlo.get_tuple_element"(%t0_1) {index = 0 : i32} : (tuple<tensor<1xi32>, tensor<2xi32>>) ->… 113 …%t1 = "mhlo.get_tuple_element"(%t0_1) {index = 1 : i32} : (tuple<tensor<1xi32>, tensor<2xi32>>) ->… [all …]
|
H A D | mhlo_mark_shape_calc_op.mlir | 12 // CHECK: mhlo.get_tuple_element 15 …%element = "mhlo.get_tuple_element"(%tuple) {index = 0 : i32} : (tuple<tensor<i64>>) -> tensor<i64> 28 // CHECK: mhlo.get_tuple_element 31 …%element = "mhlo.get_tuple_element"(%tuple) {index = 0 : i32} : (tuple<tensor<i64>>) -> tensor<i64> 46 // CHECK: mhlo.get_tuple_element 50 // CHECK: mhlo.get_tuple_element 53 …%element = "mhlo.get_tuple_element"(%tuple) {index = 0 : i32} : (tuple<tensor<i64>>) -> tensor<i64> 55 …%element2 = "mhlo.get_tuple_element"(%tuple2) {index = 0 : i32} : (tuple<tensor<i64>>) -> tensor<i…
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
H A D | while.hlotxt | 38 // CHECK-NEXT: %[[GTE_0:.*]] = "mhlo.get_tuple_element"(%[[ARG_0]]) {index = 0 39 // CHECK-NEXT: %[[GTE_1:.*]] = "mhlo.get_tuple_element"(%[[ARG_0]]) {index = 1 40 // CHECK-NEXT: %[[GTE_2:.*]] = "mhlo.get_tuple_element"(%[[ARG_0]]) {index = 2 41 // CHECK-NEXT: %[[GTE_3:.*]] = "mhlo.get_tuple_element"(%[[ARG_0]]) {index = 3 59 // CHECK-NEXT: %[[GTE_0:.*]] = "mhlo.get_tuple_element"(%[[ARG_0]]) {index = 0 60 // CHECK-NEXT: %[[GTE_1:.*]] = "mhlo.get_tuple_element"(%[[ARG_0]]) {index = 1 271 // CHECK-NEXT: %[[GTE_0:.*]] = "mhlo.get_tuple_element"(%arg0) {index = 0 : i32} : (tuple<tensor… 272 // CHECK-NEXT: %[[GTE_1:.*]] = "mhlo.get_tuple_element"(%arg0) {index = 1 : i32} : (tuple<tensor… 273 // CHECK-NEXT: %[[GTE_2:.*]] = "mhlo.get_tuple_element"(%1) {index = 0 : i32} : (tuple<tensor<i3… 274 // CHECK-NEXT: %[[GTE_3:.*]] = "mhlo.get_tuple_element"(%1) {index = 1 : i32} : (tuple<tensor<i3… [all …]
|
H A D | while.mlir | 191 …%0 = "mhlo.get_tuple_element"(%arg0) {index = 0 : i32} : (tuple<tensor<i32>, tuple<tensor<i32>, tu… 192 …%1 = "mhlo.get_tuple_element"(%arg0) {index = 1 : i32} : (tuple<tensor<i32>, tuple<tensor<i32>, tu… 193 …%2 = "mhlo.get_tuple_element"(%1) {index = 0 : i32} : (tuple<tensor<i32>, tuple<tensor<i32>>>) -> … 194 …%3 = "mhlo.get_tuple_element"(%1) {index = 1 : i32} : (tuple<tensor<i32>, tuple<tensor<i32>>>) -> … 195 %4 = "mhlo.get_tuple_element"(%3) {index = 0 : i32} : (tuple<tensor<i32>>) -> tensor<i32> 309 …%4 = "mhlo.get_tuple_element"(%3) {index = 0 : i32} : (tuple<tensor<i32>, tensor<i32>>) -> tensor<… 310 …%5 = "mhlo.get_tuple_element"(%3) {index = 1 : i32} : (tuple<tensor<i32>, tensor<i32>>) -> tensor<…
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | ir_emitter.cc | 110 Status IrEmitter::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument 111 auto operand = get_tuple_element->operand(0); in HandleGetTupleElement() 114 *get_tuple_element, in HandleGetTupleElement() 116 get_tuple_element->shape(), get_tuple_element->tuple_index(), in HandleGetTupleElement()
|
H A D | ir_emitter.h | 76 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/mhlo/ |
H A D | mhlo_flatten_tuple.mlir | 10 …%1 = "mhlo.get_tuple_element"(%0) {index = 0 : i32} : (tuple<tensor<6xf32>, tensor<3xf32>>) -> ten… 11 …%2 = "mhlo.get_tuple_element"(%0) {index = 1 : i32} : (tuple<tensor<6xf32>, tensor<3xf32>>) -> ten…
|
H A D | ops.mlir | 1860 func.func @get_tuple_element(%arg0: tuple<tensor<f32>, tensor<i32>>) -> tensor<f32> { 1861 …%0 = "mhlo.get_tuple_element"(%arg0) {index = 0 : i32} : (tuple<tensor<f32>, tensor<i32>>) -> tens… 1868 …%0 = "mhlo.get_tuple_element"(%arg0) {index = 1 : i32} : (tuple<tensor<f32>, !mhlo.token>) -> !mhl… 1876 …%0 = "mhlo.get_tuple_element"(%arg0) {index = 0 : i32} : (tuple<tensor<f32>, tensor<i32>>) -> tens… 1884 …%0 = "mhlo.get_tuple_element"(%arg0) {index = 2 : i32} : (tuple<tensor<f32>, tensor<i32>>) -> tens… 3747 …%t0 = "mhlo.get_tuple_element"(%arg2) {index = 0 : i32} : (tuple<tensor<2xi32>, tuple<tensor<1xf32… 3754 …%t0 = "mhlo.get_tuple_element"(%arg2) {index = 0 : i32} : (tuple<tensor<2xi32>, tuple<tensor<1xf32… 3755 …%t1_2 = "mhlo.get_tuple_element"(%arg2) {index = 1 : i32} : (tuple<tensor<2xi32>, tuple<tensor<1xf… 3756 …%t1 = "mhlo.get_tuple_element"(%t1_2) {index = 0 : i32} : (tuple<tensor<1xf32>, tensor<3xf32>>) ->… 3757 …%t2 = "mhlo.get_tuple_element"(%t1_2) {index = 1 : i32} : (tuple<tensor<1xf32>, tensor<3xf32>>) ->… [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | tuple_points_to_analysis.cc | 224 HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument 227 int64_t element_index = get_tuple_element->tuple_index(); in HandleGetTupleElement() 229 PointsToSet& points_to_set = CreateEmptyPointsToSet(get_tuple_element); in HandleGetTupleElement() 231 *PerInst(get_tuple_element->operand(0))->points_to_set; in HandleGetTupleElement()
|
H A D | dfs_hlo_visitor_with_default.h | 165 Status HandleGetTupleElement(HloInstructionPtr get_tuple_element) override { in HandleGetTupleElement() argument 166 return DefaultAction(get_tuple_element); in HandleGetTupleElement()
|
H A D | logical_buffer_analysis.h | 61 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
H A D | tuple_points_to_analysis.h | 248 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
H A D | hlo_cost_analysis.h | 94 const HloInstruction* get_tuple_element) override;
|
H A D | hlo_evaluator.cc | 3006 Status HloEvaluator::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument 3007 const auto result_shape = get_tuple_element->shape(); in HandleGetTupleElement() 3008 const int64_t index = get_tuple_element->tuple_index(); in HandleGetTupleElement() 3010 auto operand = get_tuple_element->operand(0); in HandleGetTupleElement() 3021 evaluated_[get_tuple_element] = in HandleGetTupleElement() 3023 return evaluated_[get_tuple_element].CopyFrom(operand_tuple_literal, in HandleGetTupleElement()
|
H A D | hlo_verifier.cc | 997 Status ShapeVerifier::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument 998 return CheckShape(get_tuple_element, in HandleGetTupleElement() 1000 get_tuple_element->operand(0)->shape(), in HandleGetTupleElement() 1001 get_tuple_element->tuple_index())); in HandleGetTupleElement()
|
H A D | hlo_evaluator.h | 299 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
H A D | algebraic_simplifier.h | 291 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
H A D | hlo_verifier.h | 177 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
H A D | legalize_tf_control_flow.cc | 252 auto get_tuple_element = builder->create<mhlo::GetTupleElementOp>( in ReplaceImplicitInputsWithTupleElements() local 255 implicit_input_elements.emplace_back(get_tuple_element.getResult()); in ReplaceImplicitInputsWithTupleElements() 259 use.set(get_tuple_element.getResult()); in ReplaceImplicitInputsWithTupleElements()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
H A D | legalize-tf-communication.mlir | 506 // CHECK-DAG: [[IF_ARG_ELEMENT0:%.*]] = "mhlo.get_tuple_element"([[ARG1]]) {index = 0 507 // CHECK-DAG: [[IF_ARG_ELEMENT1:%.*]] = "mhlo.get_tuple_element"([[ARG1]]) {index = 1 508 …%0 = "mhlo.get_tuple_element"(%arg1) {index = 0 : i32} : (tuple<tensor<f32>, tensor<f32>>) -> tens… 509 …%1 = "mhlo.get_tuple_element"(%arg1) {index = 1 : i32} : (tuple<tensor<f32>, tensor<f32>>) -> tens… 527 …%0 = "mhlo.get_tuple_element"(%arg1) {index = 0 : i32} : (tuple<tensor<f32>, tensor<f32>>) -> tens… 528 …%1 = "mhlo.get_tuple_element"(%arg1) {index = 1 : i32} : (tuple<tensor<f32>, tensor<f32>>) -> tens…
|
H A D | mlir_hlo_builder_test.cc | 152 …R"("mhlo.get_tuple_element"(%2) {index = 1 : i32} : (tuple<tensor<3x7xf32>, tensor<f32>>) -> tenso… in TEST_F()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/mhlo/canonicalize/ |
H A D | tuple.mlir | 8 …%element = "mhlo.get_tuple_element"(%tuple) {index = 0 : i32} : (tuple<tensor<i32>>) -> tensor<i32>
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
H A D | ops.mlir | 737 …%0 = "mhlo.get_tuple_element"(%arg0) {index = 0 : i32} : (tuple<tuple<tensor<f32>>, tensor<f32>>) … 738 %1 = "mhlo.get_tuple_element"(%0) {index = 0 : i32} : (tuple<tensor<f32>>) -> tensor<f32> 739 …%2 = "mhlo.get_tuple_element"(%arg0) {index = 1 : i32} : (tuple<tuple<tensor<f32>>, tensor<f32>>) … 740 %3 = "mhlo.get_tuple_element"(%arg1) {index = 0 : i32} : (tuple<tensor<f32>>) -> tensor<f32>
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
H A D | ir_emitter.h | 145 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
H A D | ir_emitter.cc | 383 Status IrEmitter::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument 388 const HloInstruction* operand = get_tuple_element->operand(0); in HandleGetTupleElement() 389 const Shape& shape = get_tuple_element->shape(); in HandleGetTupleElement() 390 emitted_value_[get_tuple_element] = llvm_ir::EmitGetTupleElement( in HandleGetTupleElement() 391 shape, get_tuple_element->tuple_index(), MinimumAlignmentForShape(shape), in HandleGetTupleElement()
|