| /external/tensorflow/tensorflow/compiler/xla/ |
| D | primitive_util.cc | 82 bool IsComplexType(PrimitiveType type) { return type == C64 || type == C128; } in IsComplexType()
119 case C64: in BitWidth()
168 case C64: in ComplexComponentType()
|
| D | literal_util.cc | 151 case C64: in Zero()
192 case C64: in One()
231 case C64: in MinValue()
308 case C64: { in NanValue()
402 case C64: in ReshapeSlice()
449 case C64: in GetFirstScalarLiteral()
|
| D | primitive_util.h | 131 return C64;
238 struct PrimitiveTypeToNative<C64> {
|
| D | literal.cc | 508 COPY_ELEMENTS(C64, complex64); in CopyFrom()
680 case C64: in CopySliceFrom()
956 case C64: in Slice()
1003 case C64: { in GetAsString()
1073 case C64: in GetAsComplex128()
1459 case C64: in ConvertIfDestTypeMatches()
1463 return ConvertIfTypesMatch<primitive_src_type, C64>(src_literal, false); in ConvertIfDestTypeMatches()
1663 case C64: in EqualElements()
1817 case C64: in IsAllComplex()
1886 case C64: { in IsAllFirst()
[all …]
|
| D | comparison_util.cc | 109 case C64: in DefaultComparisonType()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | unary_op_test.cc | 44 if (primitive_util::NativeToPrimitiveType<T>() == C64) { in AbsSize0TestHelper()
130 Sub(Mul(sign, ConvertElementType(abs, C64)), arg); in SignAbsTestHelper()
174 Add(Add(sgnf0, sgnf), ConvertElementType(sgni, F32)), C64)); in XLA_TEST_F()
|
| D | exhaustive_op_test_utils.h | 101 static constexpr bool kIsComplex = (T == C128 || T == C64);
113 : T == C128 ? F64 : T == C64 ? F32 : PRIMITIVE_TYPE_INVALID;
122 : RefT::value == C64 ? F32 : PRIMITIVE_TYPE_INVALID;
131 : (T == C64 || T == F32)
958 inline ErrorSpec DefaultSpecGenerator<C64, 1>(complex64) {
|
| D | tuple_test.cc | 462 Shape c64r0 = ShapeUtil::MakeShape(C64, {}); in XLA_TEST_F()
463 Shape c64r1 = ShapeUtil::MakeShape(C64, {2}); in XLA_TEST_F()
464 Shape c64r2 = ShapeUtil::MakeShape(C64, {3, 2}); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | types.cc | 42 {{'c', 8}, C64}, in DtypeToPrimitiveType()
83 case C64: in PrimitiveTypeToDtype()
123 case C64: in FormatDescriptorForPrimitiveType()
163 case C64: in TypeDescriptorForPrimitiveType()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | type_util.cc | 71 *type = xla::C64; in DataTypeToPrimitiveType()
91 {xla::C64, DT_COMPLEX64}, in EncodePrimitiveTypeAsDataType()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | util.cc | 54 case xla::C64: in FloatLiteral()
99 case xla::C64: in IntegerLiteral()
|
| /external/llvm-project/clang/test/CodeGenOpenCL/ |
| D | shifts.cl | 37 //OPT-NEXT: [[C64:%.+]] = ashr i64 %a, [[M64]]
40 //OPT-NEXT: [[E64:%.+]] = add nsw i64 [[C64]], 4
|
| /external/clang/test/CodeGenOpenCL/ |
| D | shifts.cl | 37 //OPT-NEXT: [[C64:%.+]] = ashr i64 %a, [[M64]]
40 //OPT-NEXT: [[E64:%.+]] = add nsw i64 [[C64]], 4
|
| /external/llvm-project/mlir/test/Dialect/Linalg/ |
| D | fusion-pattern.mlir | 26 // CHECK-DAG: %[[C64:.+]] = constant 64 : index
35 // CHECK-SAME: step (%[[C32]], %[[C64]]) {
97 // CHECK-DAG: %[[C64:.+]] = constant 64 : index
104 // CHECK-SAME: (%[[C0]]) to (%[[N]]) step (%[[C64]]) {
175 // CHECK-DAG: %[[C64:.+]] = constant 64 : index
207 // CHECK-SAME: (%[[C0]]) to (%[[N_2]]) step (%[[C64]]) {
258 // CHECK-DAG: %[[C64:.+]] = constant 64 : index
288 // CHECK-SAME: (%[[C0]]) to (%[[N_2]]) step (%[[C64]]) {
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | type_to_shape_test.cc | 200 Shape shape = ShapeUtil::MakeShapeWithLayout(PrimitiveType::C64, {2, 4, 3, 3}, in TEST()
211 converted, ShapeUtil::MakeShapeWithLayout(PrimitiveType::C64, in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | driver.cc | 96 C64, enumerator
381 case C64: in Fill()
426 case C64: in Display()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/ |
| D | tpu_driver.cc | 64 case xla::PrimitiveType::C64: in ByteSizeOfPrimitiveType()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 107 case C64: in DoGemmWithAlgorithm()
293 case C64: in RunGemm()
|
| D | cholesky_thunk.cc | 94 case C64: { in ExecuteOnStream()
|
| D | triangular_solve_thunk.cc | 116 case C64: { in ExecuteOnStream()
|
| D | cusolver_context.cc | 154 case C64: { in PotrfBufferSize()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 203 typed_visitors_[C64] = in HloEvaluator()
544 case C64: in HandleIsFinite()
607 case C64: { in HandleReal()
653 case C64: { in HandleImag()
684 case C64: { in HandleComplex()
788 case C64: { in HandleCompare()
1280 if (fft_type != FftType::RFFT && input_elt_type != PrimitiveType::C64) { in CheckParameters()
1282 input_elt_type, PrimitiveType::C64); in CheckParameters()
1299 if (fft_type != FftType::IRFFT && output_elt_type != PrimitiveType::C64) { in CheckParameters()
1301 output_elt_type, PrimitiveType::C64); in CheckParameters()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | constants.h | 59 case C64: in ConstantR0WithType()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 771 case C64: in EmitCallToRuntime()
775 float_type = llvm_ir::PrimitiveTypeToIrType(C64, module); in EmitCallToRuntime()
952 case C64: in AreGemmShapes()
1003 dot_info.result_shape.element_type() == C64 || in CanEmitTiledLlvmIrGemm()
1050 C64 == type || C128 == type); in EmitNonBatchDotOperation()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reduction_ops.cc | 92 if (xla_reduction_type == xla::C64 || xla_reduction_type == xla::C128 || in PrimitiveTypeCheck()
|