| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduction_ops_gpu_complex128.cu.cc | 54 DEFINE_FOR_TYPE_AND_R(complex128, Eigen::internal::SumReducer<complex128>);
55 DEFINE_FOR_TYPE_AND_R(complex128, functor::MeanReducer<complex128>);
56 DEFINE_FOR_TYPE_AND_R(complex128, functor::EuclideanNormReducer<complex128>);
57 DEFINE_FOR_TYPE_AND_R(complex128, Eigen::internal::ProdReducer<complex128>);
|
| D | determinant_op_gpu.cu.cc | 68 __device__ inline complex128 operator*(const complex128& a, in operator *()
69 const complex128& b) { in operator *()
72 __device__ inline complex128 operator*(const complex128& a, const double& b) { in operator *()
73 return complex128(Eigen::numext::real(a) * b, Eigen::numext::imag(a) * b); in operator *()
75 __device__ inline complex128 operator/(const complex128& a, const double& b) { in operator /()
144 template struct DeterminantFromPivotedLUFunctor<GPUDevice, complex128>;
166 template struct LogDeterminantFromPivotedLUFunctor<GPUDevice, complex128>;
|
| D | self_adjoint_eig_v2_op_complex128.cc | 20 REGISTER_LINALG_OP("SelfAdjointEigV2", (SelfAdjointEigV2Op<complex128>),
21 complex128);
24 REGISTER_LINALG_OP("BatchSelfAdjointEigV2", (SelfAdjointEigV2Op<complex128>),
25 complex128);
|
| D | svd_op_complex128.cc | 20 REGISTER_LINALG_OP("Svd", (SvdOp<complex128>), complex128);
21 REGISTER_LINALG_OP("BatchSvd", (SvdOp<complex128>), complex128);
|
| D | qr_op_complex128.cc | 20 REGISTER_LINALG_OP("Qr", (QrOp<complex128>), complex128);
29 .TypeConstraint<complex128>("T")
33 QrOp<complex128>);
|
| D | cwise_op_div.cc | 20 bfloat16, complex64, complex128);
26 bfloat16, complex64, complex128);
28 double, complex64, complex128);
32 uint16, int16, int64, complex64, complex128);
36 complex64, complex128);
|
| D | cwise_op_add_2.cc | 26 complex128, string);
30 complex128);
33 complex128);
35 complex128);
|
| D | cwise_op_reciprocal.cc | 20 complex64, complex128);
27 Eigen::half, double, complex64, complex128);
34 double, complex64, complex128);
44 Eigen::half, double, complex64, complex128);
|
| D | determinant_op.cc | 418 REGISTER_LINALG_OP_GPU("MatrixDeterminant", (DeterminantOpGpu<complex128>),
419 complex128);
428 (LogDeterminantOpGpu<complex128>), complex128);
434 REGISTER_LINALG_OP("MatrixDeterminant", (DeterminantOp<complex128>),
435 complex128);
440 REGISTER_LINALG_OP("BatchMatrixDeterminant", (DeterminantOp<complex128>),
441 complex128);
447 REGISTER_LINALG_OP("LogMatrixDeterminant", (LogDeterminantOp<complex128>),
448 complex128);
|
| D | cwise_op_conj.cc | 21 REGISTER2(UnaryOp, CPU, "Conj", functor::conj, complex64, complex128);
33 Name("Conj").Device(DEVICE_GPU).TypeConstraint<complex128>("T"),
34 UnaryOp<GPUDevice, functor::conj<complex128>>);
|
| D | cwise_op_xlogy.cc | 20 complex64, complex128);
31 REGISTER_SYCL_KERNEL(complex128);
38 complex64, complex128);
|
| D | mkl_transpose_op.cc | 89 Status MKLTranspose2D<complex128>(const char trans, const Tensor& in, in MKLTranspose2D()
94 reinterpret_cast<const MKL_Complex16*>(in.flat<complex128>().data()), in MKLTranspose2D()
97 const_cast<complex128*>(out->flat<complex128>().data())), in MKLTranspose2D()
173 return MKLTranspose2D<complex128>(kMKLTranspose, in, out); in DoTranspose()
217 return MKLTranspose2D<complex128>(kMKLConjugateTranspose, in, out); in DoTranspose()
|
| D | cholesky_op.cc | 173 REGISTER_LINALG_OP_GPU("Cholesky", (CholeskyOpGpu<complex128>), complex128);
180 REGISTER_LINALG_OP("Cholesky", (CholeskyOp<complex128>), complex128);
|
| D | parse_tensor_test.cc | 170 MakeOp<complex128>(TensorShape({3}), [](int x) -> complex128 { in TEST_F()
171 return complex128{x / 3., x / 2.}; in TEST_F()
175 ParseSerializedOutput<complex128>(GetOutput(0), &parse_output); in TEST_F()
176 test::ExpectTensorEqual<complex128>(parse_output, GetInput(0)); in TEST_F()
|
| D | matrix_solve_op.cc | 363 REGISTER_LINALG_OP_GPU("MatrixSolve", (MatrixSolveOpGpu<complex128>),
364 complex128);
371 REGISTER_LINALG_OP("MatrixSolve", (MatrixSolveOp<complex128>), complex128);
375 REGISTER_LINALG_OP("BatchMatrixSolve", (MatrixSolveOp<complex128>), complex128);
|
| D | tile_functor_gpu_complex128.cu.cc | 26 template struct Tile<GpuDevice, complex128, int32>;
27 template struct Tile<GpuDevice, complex128, int64>;
|
| D | cwise_op_equal_to_2.cc | 26 complex128, string, bool);
29 complex64, complex128, bool);
|
| D | cwise_op_not_equal_to_2.cc | 26 complex64, complex128, string, bool);
29 complex64, complex128, bool);
|
| D | cwise_op_mul_2.cc | 26 complex64, complex128);
29 complex64, complex128);
|
| D | cwise_op_exp.cc | 20 complex64, complex128);
24 complex64, complex128);
|
| D | cwise_op_real.cc | 28 REGISTER_COMPLEX(CPU, double, complex128);
32 REGISTER_COMPLEX(GPU, double, complex128);
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | linalg_ops_test.py | 46 if dtype in [np.complex64, np.complex128]:
82 (np.complex64, 0.05), (np.complex128, 1e-5)]:
95 (np.complex64, 0.05), (np.complex128, 1e-5)]:
112 (np.complex64, 0.05), (np.complex128, 1e-5)]:
123 (np.complex64, 0.05), (np.complex128, 1e-5)]:
136 for dtype in np.float64, np.float64, np.complex64, np.complex128:
207 dtypes.complex128
236 dtypes.complex128
|
| D | determinant_op_test.py | 116 np.array([[2., 3.], [3., 4.]]).astype(np.complex128))
118 np.array([[0., 0.], [0., 0.]]).astype(np.complex128))
121 np.complex128))
126 ], [1., 6., 7., 4., 7.], [2., 3., 4., 5., 6.]]).astype(np.complex128))
129 np.random.rand(3, 4, 5, 2, 2).astype(np.complex128))
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | dtypes.py | 117 elif base == complex128:
157 return self.base_dtype in (complex64, complex128)
188 self.base_dtype in (bool, string, complex64, complex128)):
212 self.base_dtype in (bool, string, complex64, complex128)):
355 complex128 = DType(types_pb2.DT_COMPLEX128) variable
420 types_pb2.DT_COMPLEX128: complex128,
548 np.complex128: complex128,
604 np.complex128,
646 np.complex128,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_test.cc | 103 EXPECT_FALSE(std::is_trivial<complex128>::value); in TEST()
1014 t.flat<complex128>().setRandom(); in TEST()
1015 TestCopies<complex128>(t); in TEST()
1069 Tensor t1 = test::AsTensor<complex128>({0, in TEST()
1071 complex128(2), in TEST()
1072 complex128(3, 3), in TEST()
1073 complex128(0, 4), in TEST()
1074 complex128(2, 5)}, in TEST()
1077 t2.flat<complex128>() = t1.flat<complex128>() * complex128(0, 2); in TEST()
1078 Tensor t3 = test::AsTensor<complex128>( in TEST()
[all …]
|