Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-control-flow.mlir11 // CHECK: [[VAL4:%.+]] = "mhlo.get_tuple_element"([[THEN_ARG]]) {index = 0 : i32}
12 // CHECK: [[VAL5:%.+]] = "mhlo.get_tuple_element"([[THEN_ARG]]) {index = 1 : i32}
18 // CHECK: [[VAL4:%.+]] = "mhlo.get_tuple_element"([[ELSE_ARG]]) {index = 0 : i32}
19 // CHECK: [[VAL5:%.+]] = "mhlo.get_tuple_element"([[ELSE_ARG]]) {index = 1 : i32}
26 // CHECK: [[VAL3:%.+]] = "mhlo.get_tuple_element"([[VAL2]]) {index = 0 : i32}
54 // CHECK: [[VAL5:%.+]] = "mhlo.get_tuple_element"([[TRUE_ARG]]) {index = 0 : i32}
62 // CHECK: [[VAL5:%.+]] = "mhlo.get_tuple_element"([[FALSE_ARG]]) {index = 0 : i32}
70 // CHECK: [[VAL4:%.+]] = "mhlo.get_tuple_element"([[VAL3]]) {index = 0 : i32}
83 …// CHECK: %[[TUPLE_ELEMENT_0:.*]] = "mhlo.get_tuple_element"(%[[TUPLE_ARG]]) {index = 0 : i32}…
84 …// CHECK: %[[TUPLE_ELEMENT_1:.*]] = "mhlo.get_tuple_element"(%[[TUPLE_ARG]]) {index = 1 : i32}…
[all …]
Dlegalize-tf-communication.mlir45 // CHECK: [[RECV_RETVAL0_VAL:%.*]] = "mhlo.get_tuple_element"([[RECV_RETVAL0_TUPLE]])
50 // CHECK: [[RECV_RETVAL0_TOKEN:%.*]] = "mhlo.get_tuple_element"([[RECV_RETVAL0_TUPLE]])
62 // CHECK: [[RECV_RETVAL1_VAL:%.*]] = "mhlo.get_tuple_element"([[RECV_RETVAL1_TUPLE]])
67 // CHECK: [[RECV_RETVAL1_TOKEN:%.*]] = "mhlo.get_tuple_element"([[RECV_RETVAL1_TUPLE]])
99 // CHECK: "mhlo.get_tuple_element"
101 // CHECK: "mhlo.get_tuple_element"
193 // CHECK: [[RECV_VAL:%.*]] = "mhlo.get_tuple_element"([[RECV_TUPLE]])
197 // CHECK: [[RECV_TOKEN:%.*]] = "mhlo.get_tuple_element"([[RECV_TUPLE]])
224 // CHECK: [[RECV0_RETVAL0_VAL:%.*]] = "mhlo.get_tuple_element"([[RECV0_RETVAL0_TUPLE]])
227 // CHECK: [[RECV0_RETVAL0_TOKEN:%.*]] = "mhlo.get_tuple_element"([[RECV0_RETVAL0_TUPLE]])
[all …]
Dmlir_hlo_builder_test.cc142 …R"("mhlo.get_tuple_element"(%2) {index = 1 : i32} : (tuple<tensor<3x7xf32>, tensor<f32>>) -> tenso… in TEST_F()
Dlegalize-tf.mlir43 …// CHECK: "mhlo.get_tuple_element"(%[[RESULT0]]) {index = 0 : i32} : (tuple<tensor<8x8x8x8xf32>, t…
44 …// CHECK: "mhlo.get_tuple_element"(%[[RESULT0]]) {index = 1 : i32} : (tuple<tensor<8x8x8x8xf32>, t…
45 …// CHECK: %[[VAR:.*]] = "mhlo.get_tuple_element"(%[[RESULT0]]) {index = 2 : i32} : (tuple<tensor<8…
75 …// CHECK: "mhlo.get_tuple_element"(%[[RESULT0]]) {index = 0 : i32} : (tuple<tensor<8x8x8x8xf32>, t…
76 …// CHECK: "mhlo.get_tuple_element"(%[[RESULT0]]) {index = 1 : i32} : (tuple<tensor<8x8x8x8xf32>, t…
77 …// CHECK: %[[VAR:.*]] = "mhlo.get_tuple_element"(%[[RESULT0]]) {index = 2 : i32} : (tuple<tensor<8…
87 …// CHECK: %[[VAR:.*]] = "mhlo.get_tuple_element"(%[[RESULT0]]) {index = 2 : i32} : (tuple<tensor<8…
96 // CHECK-DAG: %[[BATCH_MEAN:.*]] = "mhlo.get_tuple_element"(%[[RESULT0]]) {index = 1 : i32}
97 // CHECK-DAG: %[[BATCH_VAR:.*]] = "mhlo.get_tuple_element"(%[[RESULT0]]) {index = 2 : i32}
202 …// CHECK-NEXT: %[[tact:.*]] = "mhlo.get_tuple_element"(%[[training]]) {index = 0 : i32} : (tuple<t…
[all …]
/external/llvm-project/mlir/test/Transforms/
Ddecompose-call-graph-types.mlir9 // CHECK: %[[RET0:.*]] = "test.get_tuple_element"(%[[ARG_MATERIALIZED]]) {index = 0 : i32…
10 // CHECK: %[[RET1:.*]] = "test.get_tuple_element"(%[[ARG_MATERIALIZED]]) {index = 1 : i32…
49 // CHECK: %[[CALL_ARG0:.*]] = "test.get_tuple_element"(%[[ARG_MATERIALIZED]]) {index = 0 …
50 // CHECK: %[[CALL_ARG1:.*]] = "test.get_tuple_element"(%[[ARG_MATERIALIZED]]) {index = 1 …
53 // CHECK: %[[RET0:.*]] = "test.get_tuple_element"(%[[CALL_RESULT_RECOMPOSED]]) {index = 0…
54 // CHECK: %[[RET1:.*]] = "test.get_tuple_element"(%[[CALL_RESULT_RECOMPOSED]]) {index = 1…
82 // CHECK: %[[RET0:.*]] = "test.get_tuple_element"(%[[UNCONVERTED_VALUE]]) {index = 0 : i3…
83 // CHECK: %[[RET1:.*]] = "test.get_tuple_element"(%[[UNCONVERTED_VALUE]]) {index = 1 : i3…
106 // CHECK: %[[ARG_TUPLE_0:.*]] = "test.get_tuple_element"(%[[ARG_TUPLE]]) {index = 0 : i32…
107 // CHECK: %[[ARG_TUPLE_1:.*]] = "test.get_tuple_element"(%[[ARG_TUPLE]]) {index = 1 : i32…
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dif.mlir7 %0 = "mhlo.get_tuple_element"(%arg0) {index = 0 : i32} : (tuple<tensor<f32>>) -> tensor<f32>
21 %0 = "mhlo.get_tuple_element"(%arg0) {index = 0 : i32} : (tuple<tensor<f32>>) -> tensor<f32>
46 %6 = "mhlo.get_tuple_element"(%arg1) {index = 0 : i32} : (tuple<tensor<f32>>) -> tensor<f32>
52 %6 = "mhlo.get_tuple_element"(%arg1) {index = 0 : i32} : (tuple<tensor<f32>>) -> tensor<f32>
59 %3 = "mhlo.get_tuple_element"(%2) {index = 0 : i32} : (tuple<tensor<f32>>) -> tensor<f32>
Dif_conditional.hlotxt34 // CHECK: [[R7:%.+]] = "mhlo.get_tuple_element"([[A1]])
40 // CHECK: [[R7:%.+]] = "mhlo.get_tuple_element"([[A1]])
47 // CHECK: [[R4:%.+]] = "mhlo.get_tuple_element"([[R3]])
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlegalize_to_scf.mlir11 …%2 = "mhlo.get_tuple_element"(%arg9) {index = 0 : i32} : (tuple<tensor<i32>, tensor<i32>, tensor<i…
12 …%3 = "mhlo.get_tuple_element"(%arg9) {index = 2 : i32} : (tuple<tensor<i32>, tensor<i32>, tensor<i…
17 …%2 = "mhlo.get_tuple_element"(%arg9) {index = 0 : i32} : (tuple<tensor<i32>, tensor<i32>, tensor<i…
19 …%4 = "mhlo.get_tuple_element"(%arg9) {index = 1 : i32} : (tuple<tensor<i32>, tensor<i32>, tensor<i…
20 …%5 = "mhlo.get_tuple_element"(%arg9) {index = 2 : i32} : (tuple<tensor<i32>, tensor<i32>, tensor<i…
Dsink-constants-to-control-flow.mlir44 %3 = "mhlo.get_tuple_element"(%arg1) {index = 0 : i32} : (tuple<tensor<i64>>) -> tensor<i64>
52 %6 = "mhlo.get_tuple_element"(%arg1) {index = 0 : i32} : (tuple<tensor<i64>>) -> tensor<i64>
58 %9 = "mhlo.get_tuple_element"(%2) {index = 0 : i32} : (tuple<tensor<i64>>) -> tensor<i64>
Dtuple.mlir8 …%element = "mhlo.get_tuple_element"(%tuple) {index = 0 : i32} : (tuple<tensor<i32>>) -> tensor<i32>
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter.cc182 Status IrEmitter::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
183 auto operand = get_tuple_element->operand(0); in HandleGetTupleElement()
186 *get_tuple_element, in HandleGetTupleElement()
188 get_tuple_element->shape(), get_tuple_element->tuple_index(), in HandleGetTupleElement()
Dir_emitter.h80 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dfused_ir_emitter.h83 Status HandleGetTupleElement(const HloInstruction* get_tuple_element);
Dfused_ir_emitter.cc112 const HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_control_flow.cc201 auto get_tuple_element = builder->create<mhlo::GetTupleElementOp>( in ReplaceImplicitInputs() local
204 implicit_input_elements.emplace_back(get_tuple_element.getResult()); in ReplaceImplicitInputs()
208 use.set(get_tuple_element.getResult()); in ReplaceImplicitInputs()
Dlegalize_tf_communication.cc294 auto get_tuple_element = in CreateRecvOp() local
296 if (tpu_core) SetOpSharding(get_tuple_element, *tpu_core); in CreateRecvOp()
298 result.replaceAllUsesWith(get_tuple_element); in CreateRecvOp()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis.cc223 HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
226 int64 element_index = get_tuple_element->tuple_index(); in HandleGetTupleElement()
228 PointsToSet& points_to_set = CreateEmptyPointsToSet(get_tuple_element); in HandleGetTupleElement()
230 *PerInst(get_tuple_element->operand(0))->points_to_set; in HandleGetTupleElement()
Dlogical_buffer_analysis.h61 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
Ddfs_hlo_visitor_with_default.h152 Status HandleGetTupleElement(HloInstructionPtr get_tuple_element) override { in HandleGetTupleElement() argument
153 return DefaultAction(get_tuple_element); in HandleGetTupleElement()
Dtuple_points_to_analysis.h249 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
Dhlo_cost_analysis.h58 const HloInstruction* get_tuple_element) override;
Dhlo_evaluator.h227 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
Dhlo_verifier.h76 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
Dhlo_evaluator.cc1851 Status HloEvaluator::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
1852 const auto result_shape = get_tuple_element->shape(); in HandleGetTupleElement()
1853 const int64 index = get_tuple_element->tuple_index(); in HandleGetTupleElement()
1855 auto operand = get_tuple_element->operand(0); in HandleGetTupleElement()
1866 evaluated_[get_tuple_element] = in HandleGetTupleElement()
1868 return evaluated_[get_tuple_element].CopyFrom(operand_tuple_literal, in HandleGetTupleElement()
Dhlo_verifier.cc634 Status ShapeVerifier::HandleGetTupleElement(HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
635 return CheckShape(get_tuple_element, in HandleGetTupleElement()
637 get_tuple_element->operand(0)->shape(), in HandleGetTupleElement()
638 get_tuple_element->tuple_index())); in HandleGetTupleElement()

12