| /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...] |
