/external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
D | legalize-tf-control-flow.mlir | 11 // 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 …]
|
D | legalize-tf-communication.mlir | 45 // 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 …]
|
D | mlir_hlo_builder_test.cc | 142 …R"("mhlo.get_tuple_element"(%2) {index = 1 : i32} : (tuple<tensor<3x7xf32>, tensor<f32>>) -> tenso… in TEST_F()
|
D | legalize-tf.mlir | 43 …// 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/ |
D | decompose-call-graph-types.mlir | 9 // 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/ |
D | if.mlir | 7 %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>
|
D | if_conditional.hlotxt | 34 // 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/ |
D | legalize_to_scf.mlir | 11 …%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…
|
D | sink-constants-to-control-flow.mlir | 44 %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>
|
D | tuple.mlir | 8 …%element = "mhlo.get_tuple_element"(%tuple) {index = 0 : i32} : (tuple<tensor<i32>>) -> tensor<i32>
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | ir_emitter.cc | 182 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()
|
D | ir_emitter.h | 80 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
D | fused_ir_emitter.h | 83 Status HandleGetTupleElement(const HloInstruction* get_tuple_element);
|
D | fused_ir_emitter.cc | 112 const HloInstruction* get_tuple_element) { in HandleGetTupleElement() argument
|
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
D | legalize_tf_control_flow.cc | 201 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()
|
D | legalize_tf_communication.cc | 294 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/ |
D | tuple_points_to_analysis.cc | 223 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()
|
D | logical_buffer_analysis.h | 61 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
D | dfs_hlo_visitor_with_default.h | 152 Status HandleGetTupleElement(HloInstructionPtr get_tuple_element) override { in HandleGetTupleElement() argument 153 return DefaultAction(get_tuple_element); in HandleGetTupleElement()
|
D | tuple_points_to_analysis.h | 249 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
D | hlo_cost_analysis.h | 58 const HloInstruction* get_tuple_element) override;
|
D | hlo_evaluator.h | 227 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
D | hlo_verifier.h | 76 Status HandleGetTupleElement(HloInstruction* get_tuple_element) override;
|
D | hlo_evaluator.cc | 1851 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()
|
D | hlo_verifier.cc | 634 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()
|