/external/tensorflow/tensorflow/compiler/xla/ |
primitive_util.cc | 46 bool IsComplexType(PrimitiveType type) { return type == C64 || type == C128; } 83 case C64: 117 case C64:
|
primitive_util.h | 130 return C64; 235 struct PrimitiveTypeToNative<C64> {
|
literal_util.cc | 131 case C64: 168 case C64: 206 case C64: 207 LOG(FATAL) << "C64 element type has no minimum value"; 350 case C64: 397 case C64:
|
literal.cc | 414 COPY_ELEMENTS(C64, complex64); 550 case C64: 772 case C64: 820 case C64: { [all...] |
xla_data.proto | 59 C64 = 15; // Paired F32 (real, imag), as in std::complex<float>.
|
literal_test.cc | 119 EXPECT_EQ("c64[] (3.14, 2.78)", c64_lit.ToString()); 851 Literal output(ShapeUtil::MakeShape(C64, {1})); 865 Literal output(ShapeUtil::MakeShape(C64, {2, 2})); 918 Literal output(ShapeUtil::MakeShape(C64, {2, 2})); 1339 auto c64 = LiteralUtil::CreateR4WithLayout<complex64>({{ local [all...] |
shape_util_test.cc | 296 EXPECT_EQ(8, ShapeUtil::ByteSizeOfPrimitiveType(C64)); 297 EXPECT_EQ(8, ShapeUtil::ByteSizeOf(ShapeUtil::MakeShape(C64, {}))); 298 EXPECT_EQ(1600, ShapeUtil::ByteSizeOf(ShapeUtil::MakeShape(C64, {10, 20}))); [all...] |
literal_comparison.cc | 711 case C64: 799 case C64: [all...] |
shape_util.cc | 308 case C64: 554 case C64: [all...] |
/external/tensorflow/tensorflow/compiler/tf2xla/ |
type_util.cc | 70 *type = xla::C64;
|
/external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
util.cc | 54 case xla::C64: 93 case xla::C64:
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
unary_op_test.cc | 44 if (primitive_util::NativeToPrimitiveType<T>() == C64) { 130 Sub(Mul(sign, ConvertElementType(abs, C64)), arg); 174 Add(Add(sgnf0, sgnf), ConvertElementType(sgni, F32)), C64));
|
tuple_test.cc | 462 Shape c64r0 = ShapeUtil::MakeShape(C64, {}); 463 Shape c64r1 = ShapeUtil::MakeShape(C64, {2}); 464 Shape c64r2 = ShapeUtil::MakeShape(C64, {3, 2});
|
convert_test.cc | 492 ConvertElementType(ConstantR1<complex64>(&builder, x), C64);
|
test_utils.cc | 248 case C64:
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
cholesky_thunk.cc | 98 case C64: {
|
gemm_thunk.cc | 207 case C64: 224 case C64: 240 case C64: 262 case C64:
|
triangular_solve_thunk.cc | 115 case C64: {
|
cusolver_rewriter.cc | 85 case C64: {
|
ir_emission_utils.cc | 56 output_primitive_type == F64 || output_primitive_type == C64 ||
|
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
constants.h | 57 case C64:
|
/external/u-boot/include/ |
div64.h | 230 # error do_div() does not yet support the C64
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
hlo_evaluator.cc | 201 typed_visitors_[C64] = 507 case C64: 570 case C64: { 616 case C64: { 647 case C64: { 751 case C64: { [all...] |
/external/tensorflow/tensorflow/compiler/xla/python/ |
numpy_bridge.cc | 59 case C64: 97 return C64;
|
xla_client.py | 364 C64 = 15 384 PrimitiveType.C64: np.dtype('complex64'), [all...] |