| /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 /()
143 template struct DeterminantFromPivotedLUFunctor<GPUDevice, complex128>;
165 template struct LogDeterminantFromPivotedLUFunctor<GPUDevice, complex128>;
|
| D | cwise_op_div.cc | 20 bfloat16, complex64, complex128);
26 bfloat16, complex64, complex128);
28 double, complex64, complex128);
33 uint16, int16, int64, complex64, complex128);
37 complex64, complex128);
39 double, complex64, complex128);
|
| D | svd_op_complex128.cc | 20 REGISTER_LINALG_OP("Svd", (SvdOp<complex128>), complex128);
21 REGISTER_LINALG_OP("BatchSvd", (SvdOp<complex128>), 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 | qr_op_complex128.cc | 20 REGISTER_LINALG_OP("Qr", (QrOp<complex128>), complex128);
29 .TypeConstraint<complex128>("T")
33 QrOp<complex128>);
|
| D | cwise_op_reciprocal.cc | 20 complex64, complex128);
27 Eigen::half, double, complex64, complex128);
37 double, complex64, complex128, bfloat16);
40 double, complex64, complex128);
51 Eigen::half, double, complex64, complex128);
|
| D | cwise_op_add_2.cc | 26 complex128, tstring);
30 complex128);
33 complex128);
35 complex128);
|
| D | eig_op_complex128.cc | 20 REGISTER_LINALG_OP("Eig", (EigOp<complex128, complex128>), complex128);
|
| D | cwise_op_xlog1py.cc | 20 double, complex64, complex128);
31 REGISTER_SYCL_KERNEL(complex128);
38 double, complex64, 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 | 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 | 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()
220 return MKLTranspose2D<complex128>(kMKLConjugateTranspose, in, out); in DoTranspose()
|
| D | tile_functor_gpu_complex128.cu.cc | 26 template struct Tile<GpuDevice, complex128, int32>;
27 template struct Tile<GpuDevice, complex128, int64>;
|
| D | parse_tensor_test.cc | 173 MakeOp<complex128>(TensorShape({3}), [](int x) -> complex128 { in TEST_F()
174 return complex128{x / 3., x / 2.}; in TEST_F()
178 ParseSerializedOutput<complex128>(GetOutput(0), &parse_output); in TEST_F()
179 test::ExpectTensorEqual<complex128>(parse_output, GetInput(0)); in TEST_F()
|
| D | cwise_op_gpu_div.cu.cc | 23 int64, complex64, complex128);
24 DEFINE_BINARY5(div_no_nan, Eigen::half, float, double, complex64, complex128);
|
| D | cwise_op_gpu_mul.cu.cc | 23 int32, int64, complex64, complex128);
24 DEFINE_BINARY5(mul_no_nan, Eigen::half, float, double, complex64, complex128);
|
| D | cwise_op_not_equal_to_2.cc | 26 complex64, complex128, tstring, bool);
29 complex64, complex128, bool);
|
| D | cwise_op_equal_to_2.cc | 26 complex128, tstring, bool);
29 complex64, complex128, bool);
|
| D | cwise_op_mul_2.cc | 26 complex64, complex128);
29 complex64, complex128);
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | transpose_op.h | 41 struct maybe_conj_inplace<Device, complex128> {
43 functor::UnaryFunctor<Device, functor::conj<complex128>> conj;
44 conj(d, t->flat<complex128>() /*out*/,
45 const_cast<const Tensor*>(t)->flat<complex128>() /*in*/);
63 struct maybe_conj<Device, complex128> {
65 functor::UnaryFunctor<Device, functor::conj<complex128>> conj;
66 conj(d, out->flat<complex128>() /*out*/, in.flat<complex128>() /*in*/);
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | literal_test_util_test.cc | 51 LiteralUtil::CreateR0<complex128>({42.0, 64.0}), in TEST()
52 LiteralUtil::CreateR0<complex128>({64.0, 42.0}), in TEST()
82 LiteralUtil::CreateR0<complex128>({42.0, 64.0}), in TEST()
83 LiteralUtil::CreateR0<complex128>({64.0, 42.0}), in TEST()
86 LiteralUtil::CreateR0<complex128>({64.0, 42.0}), in TEST()
87 LiteralUtil::CreateR0<complex128>({42.0, 64.0}), in TEST()
90 LiteralUtil::CreateR0<complex128>({42.42, 64.0}), in TEST()
91 LiteralUtil::CreateR0<complex128>({64.0, 42.0}), in TEST()
94 LiteralUtil::CreateR0<complex128>({42.0, 64.0}), in TEST()
95 LiteralUtil::CreateR0<complex128>({64.0, 42.42}), in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 140 StatusOr<Literal> Compare<complex128>(const Shape& shape, in Compare()
144 std::function<bool(complex128, complex128)> compare_op; in Compare()
147 compare_op = [](complex128 lhs_el, complex128 rhs_el) { in Compare()
152 compare_op = [](complex128 lhs_el, complex128 rhs_el) { in Compare()
164 return compare_op(lhs_literal.Get<complex128>(multi_index), in Compare()
165 rhs_literal.Get<complex128>(multi_index)); in Compare()
206 absl::make_unique<HloEvaluatorTypedVisitor<complex128>>(this); in HloEvaluator()
603 auto result_or = ElementWiseUnaryOpImpl<double, complex128>( in HandleReal()
604 real, [](complex128 elem_operand) { return std::real(elem_operand); }, in HandleReal()
650 auto result_or = ElementWiseUnaryOpImpl<double, complex128>( in HandleImag()
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Bitcast.pbtxt | 16 (e.g. tf.complex64 or tf.complex128) as tf.cast() make imaginary part 0 while tf.bitcast()
23 >>> equality_bitcast = tf.bitcast(a, tf.complex128)
27 >>> equality_cast = tf.cast(a, tf.complex128)
29 tf.Tensor([1.+0.j 2.+0.j 3.+0.j], shape=(3,), dtype=complex128)
|