Home
last modified time | relevance | path

Searched refs:get_tuple_element (Results 1 – 25 of 33) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
H A Dmhlo_flatten_tuple.mlir64 …%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 Dmhlo_mark_shape_calc_op.mlir12 // 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 Dwhile.hlotxt38 // 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 Dwhile.mlir191 …%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 Dir_emitter.cc110 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 Dir_emitter.h76 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/mhlo/
H A Dmhlo_flatten_tuple.mlir10 …%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 Dops.mlir1860 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 Dtuple_points_to_analysis.cc224 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 Ddfs_hlo_visitor_with_default.h165 Status HandleGetTupleElement(HloInstructionPtr get_tuple_element) override { in HandleGetTupleElement() argument
166 return DefaultAction(get_tuple_element); in HandleGetTupleElement()
H A Dlogical_buffer_analysis.h61 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
H A Dtuple_points_to_analysis.h248 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
H A Dhlo_cost_analysis.h94 const HloInstruction* get_tuple_element) override;
H A Dhlo_evaluator.cc3006 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 Dhlo_verifier.cc997 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 Dhlo_evaluator.h299 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
H A Dalgebraic_simplifier.h291 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
H A Dhlo_verifier.h177 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf_control_flow.cc252 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 Dlegalize-tf-communication.mlir506 // 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 Dmlir_hlo_builder_test.cc152 …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 Dtuple.mlir8 …%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 Dops.mlir737 …%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 Dir_emitter.h145 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
H A Dir_emitter.cc383 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()

12