| /external/tensorflow/tensorflow/core/kernels/ |
| qr_op_complex64.cc | 20 REGISTER_LINALG_OP("Qr", (QrOp<complex64>), complex64);
23 REGISTER_LINALG_OP_GPU("Qr", (QrOpGpu<complex64>), complex64);
|
| self_adjoint_eig_v2_op_complex64.cc | 20 REGISTER_LINALG_OP("SelfAdjointEigV2", (SelfAdjointEigV2Op<complex64>),
21 complex64);
24 REGISTER_LINALG_OP("BatchSelfAdjointEigV2", (SelfAdjointEigV2Op<complex64>),
25 complex64);
|
| svd_op_complex64.cc | 20 REGISTER_LINALG_OP("Svd", (SvdOp<complex64>), complex64);
21 REGISTER_LINALG_OP("BatchSvd", (SvdOp<complex64>), complex64);
|
| cwise_op_gpu_arg.cu.cc | 24 DEFINE_UNARY2(get_angle, complex64, complex128);
|
| cwise_op_gpu_imag.cu.cc | 22 DEFINE_UNARY2(get_imag, complex64, complex128);
|
| cwise_op_gpu_real.cu.cc | 22 DEFINE_UNARY2(get_real, complex64, complex128);
|
| matrix_solve_ls_op_complex64.cc | 21 complex64);
|
| cwise_op_gpu_abs.cu.cc | 22 DEFINE_UNARY6(abs, Eigen::half, float, double, int64, complex64, complex128);
|
| cwise_op_gpu_conj.cu.cc | 22 DEFINE_UNARY1(conj, complex64);
|
| cwise_op_gpu_exp.cu.cc | 22 DEFINE_UNARY5(exp, Eigen::half, float, double, complex64, complex128);
|
| cwise_op_gpu_sign.cu.cc | 22 DEFINE_UNARY6(sign, Eigen::half, float, double, int64, complex64, complex128);
|
| cwise_op_gpu_sub.cu.cc | 22 DEFINE_BINARY6(sub, Eigen::half, float, double, int64, complex64, complex128);
|
| cwise_op_add_2.cc | 25 REGISTER6(BinaryOp, CPU, "Add", functor::add, int8, int16, complex64, uint8,
29 REGISTER5(BinaryOp, CPU, "AddV2", functor::add, int8, int16, complex64, uint8,
32 REGISTER4(BinaryOp, GPU, "Add", functor::add, uint8, int64, complex64,
34 REGISTER4(BinaryOp, GPU, "AddV2", functor::add, uint8, int64, complex64,
|
| cwise_op_gpu_add.cu.cc | 22 DEFINE_BINARY7(add, Eigen::half, float, double, uint8, int64, complex64,
|
| cwise_op_gpu_div.cu.cc | 23 int64, complex64, complex128);
|
| cwise_op_gpu_equal_to.cu.cc | 23 complex64, complex128, bool);
|
| cwise_op_gpu_neg.cu.cc | 22 DEFINE_UNARY7(neg, Eigen::half, float, double, int32, int64, complex64,
|
| cwise_op_gpu_not_equal_to.cu.cc | 23 int64, complex64, complex128, bool);
|
| cwise_op_equal_to_2.cc | 25 REGISTER6(BinaryOp, CPU, "Equal", functor::equal_to, int32, int64, complex64,
29 complex64, complex128, bool);
|
| cwise_op_exp.cc | 20 complex64, complex128);
24 complex64, complex128);
|
| cwise_op_mul_2.cc | 26 complex64, complex128);
29 complex64, complex128);
|
| cwise_op_not_equal_to_2.cc | 26 complex64, complex128, string, bool);
29 complex64, complex128, bool);
|
| cwise_op_conj.cc | 21 REGISTER2(UnaryOp, CPU, "Conj", functor::conj, complex64, complex128);
30 Name("Conj").Device(DEVICE_GPU).TypeConstraint<complex64>("T"),
31 UnaryOp<GPUDevice, functor::conj<complex64>>);
|
| cwise_op_acosh.cc | 20 REGISTER4(UnaryOp, CPU, "Acosh", functor::acosh, float, double, complex64,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| triangular_solve_test.cc | 38 using complex64 = xla::complex64;
56 xla::Array2D<complex64> AValsLowerComplex() {
58 {complex64(3, 1), 6, 0, 0},
59 {4, complex64(7, 2), 9, 0},
60 {5, 8, complex64(10, 3), 11}};
63 xla::Array2D<complex64> AValsUpperComplex() {
64 return {{2, 3, complex64(4, 3), 5},
65 {0, 6, complex64(7, 2), 8},
66 {0, 0, complex64(9, 1), 10}
[all...] |
