Home
last modified time | relevance | path

Searched refs:TUPLE (Results 1 – 25 of 57) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Ddynamic_parameter_binding.mlir2 …late -split-input-file -mlir-hlo-to-hlo -emit-use-tuple-args %s | FileCheck %s --check-prefix=TUPLE
14 // TUPLE-LABEL: hlo_module
15 // TUPLE: dynamic_parameter_binding
16 // TUPLE-NEXT: }
33 // TUPLE-LABEL: hlo_module
34 // TUPLE: dynamic_parameter_binding
35 // TUPLE-NEXT: entries
36 // TUPLE-NEXT: dynamic_param_index: 1
37 // TUPLE-NEXT: target_param_index: 0
38 // TUPLE-NEXT: }
[all …]
Dadd.mlir2 …lir-translate -mlir-hlo-to-hlo-text -emit-use-tuple-args %s | FileCheck %s --check-prefix=TUPLE-ARG
3 …-mlir-translate -mlir-hlo-to-hlo-text -emit-return-tuple %s | FileCheck %s --check-prefix=TUPLE-RET
7 // TUPLE-ARG-LABEL: ENTRY %main.{{.*}} (arg_tuple.1: (f32[4], f32[4])) -> f32[4]
8 // TUPLE-RET-LABEL: ENTRY %main.{{.*}} (Arg_0.1: f32[4], Arg_1.2: f32[4]) -> (f32[4])
Dinput_output_aliasing.mlir2 …-hlo-to-hlo-text -emit-use-tuple-args -emit-return-tuple %s | FileCheck %s --check-prefix=TUPLE-ARG
6 // TUPLE-ARG-LABEL: ENTRY %main
7 // TUPLE-ARG: // OutputIndex {0} aliases with input 0 at {0}
Dcase.mlir74 // CHECK: ROOT %[[TUPLE:.*]] = (f32[], f32[]) tuple(f32[] %[[NEGATE]], f32[] %[[NEGATE]])
80 // CHECK: ROOT %[[TUPLE:.*]] = (f32[], f32[]) tuple(f32[] %[[COPY]], f32[] %[[COPY]])
86 // CHECK: ROOT %[[TUPLE:.*]] = (f32[], f32[]) tuple(f32[] %[[FLOOR]], f32[] %[[FLOOR]])
96 // CHECK: %[[TUPLE:.*]] = (f32[], f32[]) conditional(s32[] %[[INDEX]], f32[] %[[OPERAND_1]], f32[] …
97 // CHECK: %[[RES_1:.*]] = f32[] get-tuple-element((f32[], f32[]) %[[TUPLE]]), index=0
98 // CHECK: %[[RES_2:.*]] = f32[] get-tuple-element((f32[], f32[]) %[[TUPLE]]), index=1
Dmultiple_return_tuple.mlir2 …mlir-hlo-to-hlo-text -emit-use-tuple-args -emit-return-tuple %s | FileCheck %s --check-prefix=TUPLE
8 // TUPLE-LABEL: ENTRY %main.{{.*}} (arg_tuple.1: (s32[4])) -> (s32[4], s32[1,2,3,4])
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/
Dadd.mlir2 … -tf-input-shapes=: -emit-use-tuple-args -emit-return-tuple | FileCheck -check-prefix=TUPLE-ARGS %s
28 // TUPLE-ARGS-LABEL: HloModule main
29 // TUPLE-ARGS: ENTRY %main.{{[0-9]+}} ([[ARG_TUPLE:.*]]: (f32[], f32[])) -> (f32[]) {
30 // TUPLE-ARGS: %[[ARG_TUPLE]] = (f32[], f32[]) parameter(0)
31 // TUPLE-ARGS: [[ARG0:%.*]] = f32[] get-tuple-element((f32[], f32[]) %[[ARG_TUPLE]]), index…
32 // TUPLE-ARGS: [[ARG1:%.*]] = f32[] get-tuple-element((f32[], f32[]) %[[ARG_TUPLE]]), index…
33 // TUPLE-ARGS: [[ADD:%.*]] = f32[] add(f32[] [[ARG0]], f32[] [[ARG1]])
34 // TUPLE-ARGS: ROOT %tuple.{{[0-9]+}} = (f32[]) tuple(f32[] [[ADD]])
35 // TUPLE-ARGS: }
37 // TUPLE-ARGS: // InputMapping {0, 1}
[all …]
/external/tensorflow/tensorflow/compiler/xla/python_api/
Dxla_shape.py56 if element_type == xla_data_pb2.TUPLE:
75 return self.element_type() == xla_data_pb2.TUPLE
141 xla_data_pb2.TUPLE,
Dxla_literal.py31 if element_type == xla_data_pb2.TUPLE:
/external/llvm-project/flang/include/flang/Common/
Dtemplate.h29 template <int N, template <typename> class PREDICATE, typename TUPLE>
32 if constexpr (N >= std::tuple_size_v<TUPLE>) { in value()
34 } else if constexpr (PREDICATE<std::tuple_element_t<N, TUPLE>>::value()) { in value()
37 return SearchTypeListHelper<N + 1, PREDICATE, TUPLE>::value(); in value()
190 template <typename TUPLE>
191 using TupleToVariant = typename TupleToVariantHelper<TUPLE>::type;
/external/tensorflow/tensorflow/compiler/tf2xla/
Dsharding_util_test.cc93 if (sharding->type() == xla::OpSharding::TUPLE) { in TEST_P()
131 sharding.set_type(xla::OpSharding::TUPLE); in CreateTupleSharding()
Dsharding_util.cc40 if (sharding.type() == xla::OpSharding::TUPLE) { in AssignOpMetadataToSharding()
125 sharding.value().type() == xla::OpSharding::TUPLE) { in ParseShardingFromEdgeSource()
Dxla_jit_compiled_cpu_function_test.cc226 ASSERT_EQ(result.element_type(), xla::TUPLE); in TEST()
277 ASSERT_EQ(result.element_type(), xla::TUPLE); in TEST()
Dxla_jit_compiled_cpu_function.cc99 if (program_shape->result().element_type() != xla::TUPLE) { in Compile()
/external/tensorflow/tensorflow/compiler/xla/
Dprimitive_util.cc125 case TUPLE: in BitWidth()
179 return primitive_type != PRIMITIVE_TYPE_INVALID && primitive_type != TUPLE && in IsArrayType()
Dprimitive_util_test.cc36 expect_ok_and_equal("tuple", TUPLE); in TEST()
Dliteral_util.cc157 case TUPLE: in Zero()
198 case TUPLE: in One()
243 case TUPLE: in MinValue()
284 case TUPLE: in MaxValue()
Dshape_util.cc128 if (element_type == OPAQUE_TYPE || element_type == TUPLE) { in MakeShapeWithLayoutInternal()
367 result.set_element_type(TUPLE); in MakeTupleShape()
467 case TUPLE: in ElementIsSigned()
733 case TUPLE: in ByteSizeOfPrimitiveType()
745 if (shape.element_type() == TUPLE) { in ByteSizeOf()
762 CHECK_EQ(TUPLE, shape.element_type()); in ByteSizeOfTupleIndexTable()
785 if (shape.element_type() == TUPLE) { in ValidateShapeWithOptionalLayoutInternal()
/external/python/cpython2/Lib/test/
Dpickletester.py1193 expected_opcode = {(0, 0): pickle.TUPLE,
1194 (0, 1): pickle.TUPLE,
1195 (0, 2): pickle.TUPLE,
1196 (0, 3): pickle.TUPLE,
1197 (0, 4): pickle.TUPLE,
1200 (1, 1): pickle.TUPLE,
1201 (1, 2): pickle.TUPLE,
1202 (1, 3): pickle.TUPLE,
1203 (1, 4): pickle.TUPLE,
1209 (2, 4): pickle.TUPLE,
/external/tensorflow/tensorflow/compiler/xla/client/
Dsharding_builder.cc74 result.set_type(OpSharding::TUPLE); in Tuple()
/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
Dxla_sharding.py193 type=xla_data_pb2.OpSharding.TUPLE, tuple_shardings=tuple_shardings)
231 type=xla_data_pb2.OpSharding.TUPLE, tuple_shardings=shardings)
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dtpu_driver.cc69 case xla::PrimitiveType::TUPLE: in ByteSizeOfPrimitiveType()
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Dinfeed_op.cc133 sharding = b->sharding()->type() == xla::OpSharding::TUPLE in Compile()
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dutil.cc115 case xla::TUPLE: in IntegerLiteral()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dtuple_ops.cc66 b->CreatePointerCast(operands[i], PrimitiveTypeToIrType(TUPLE, module)); in EmitTuple()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreduction_ops.cc93 xla_reduction_type == xla::TUPLE || in PrimitiveTypeCheck()

123