| /external/tensorflow/tensorflow/core/kernels/special_math/ |
| D | special_math_op_bessel.cc | 20 REGISTER3(UnaryOp, CPU, "BesselI0", functor::bessel_i0, Eigen::half, float,
22 REGISTER3(UnaryOp, CPU, "BesselI1", functor::bessel_i1, Eigen::half, float,
24 REGISTER3(UnaryOp, CPU, "BesselI0e", functor::bessel_i0e, Eigen::half, float,
26 REGISTER3(UnaryOp, CPU, "BesselI1e", functor::bessel_i1e, Eigen::half, float,
29 REGISTER3(UnaryOp, CPU, "BesselK0", functor::bessel_k0, Eigen::half, float,
31 REGISTER3(UnaryOp, CPU, "BesselK1", functor::bessel_k1, Eigen::half, float,
33 REGISTER3(UnaryOp, CPU, "BesselK0e", functor::bessel_k0e, Eigen::half, float,
35 REGISTER3(UnaryOp, CPU, "BesselK1e", functor::bessel_k1e, Eigen::half, float,
38 REGISTER3(UnaryOp, CPU, "BesselJ0", functor::bessel_j0, Eigen::half, float,
40 REGISTER3(UnaryOp, CPU, "BesselJ1", functor::bessel_j1, Eigen::half, float,
[all …]
|
| D | special_math_op_dawsn.cc | 20 REGISTER2(UnaryOp, CPU, "Dawsn", functor::dawsn, float, double);
22 REGISTER2(UnaryOp, GPU, "Dawsn", functor::dawsn, float, double);
|
| D | special_math_op_expint.cc | 20 REGISTER2(UnaryOp, CPU, "Expint", functor::expint, float, double);
22 REGISTER2(UnaryOp, GPU, "Expint", functor::expint, float, double);
|
| D | special_math_op_fresnel_cos.cc | 20 REGISTER2(UnaryOp, CPU, "FresnelCos", functor::fresnel_cos, float, double);
22 REGISTER2(UnaryOp, GPU, "FresnelCos", functor::fresnel_cos, float, double);
|
| D | special_math_op_fresnel_sin.cc | 20 REGISTER2(UnaryOp, CPU, "FresnelSin", functor::fresnel_sin, float, double);
22 REGISTER2(UnaryOp, GPU, "FresnelSin", functor::fresnel_sin, float, double);
|
| D | special_math_op_spence.cc | 20 REGISTER2(UnaryOp, CPU, "Spence", functor::spence, float, double);
22 REGISTER2(UnaryOp, GPU, "Spence", functor::spence, float, double);
|
| /external/eigen/Eigen/src/Core/ |
| D | CwiseUnaryOp.h | 17 template<typename UnaryOp, typename XprType>
18 struct traits<CwiseUnaryOp<UnaryOp, XprType> >
22 UnaryOp(const typename XprType::Scalar&)
32 template<typename UnaryOp, typename XprType, typename StorageKind>
54 template<typename UnaryOp, typename XprType>
55 class CwiseUnaryOp : public CwiseUnaryOpImpl<UnaryOp, XprType, typename internal::traits<XprType>::…
59 …typedef typename CwiseUnaryOpImpl<UnaryOp, XprType,typename internal::traits<XprType>::StorageKind…
65 explicit CwiseUnaryOp(const XprType& xpr, const UnaryOp& func = UnaryOp())
75 const UnaryOp& functor() const { return m_functor; }
89 const UnaryOp m_functor;
[all …]
|
| /external/eigen/Eigen/src/SparseCore/ |
| D | SparseCwiseUnaryOp.h | 17 template<typename UnaryOp, typename ArgType>
18 struct unary_evaluator<CwiseUnaryOp<UnaryOp,ArgType>, IteratorBased>
19 : public evaluator_base<CwiseUnaryOp<UnaryOp,ArgType> >
22 typedef CwiseUnaryOp<UnaryOp, ArgType> XprType;
27 CoeffReadCost = evaluator<ArgType>::CoeffReadCost + functor_traits<UnaryOp>::Cost,
33 EIGEN_INTERNAL_CHECK_COST_VALUE(functor_traits<UnaryOp>::Cost);
44 const UnaryOp m_functor;
48 template<typename UnaryOp, typename ArgType>
49 class unary_evaluator<CwiseUnaryOp<UnaryOp,ArgType>, IteratorBased>::InnerIterator
50 : public unary_evaluator<CwiseUnaryOp<UnaryOp,ArgType>, IteratorBased>::EvalIterator
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_op_abs.cc | 22 REGISTER8(UnaryOp, CPU, "Abs", functor::abs, Eigen::half, bfloat16, float,
25 REGISTER(UnaryOp, CPU, "Abs", functor::abs, bfloat16);
28 REGISTER2(UnaryOp, CPU, "ComplexAbs", functor::abs, complex64, complex128);
32 REGISTER4(UnaryOp, GPU, "Abs", functor::abs, Eigen::half, float, double, int64);
33 REGISTER2(UnaryOp, GPU, "ComplexAbs", functor::abs, complex64, complex128);
44 UnaryOp<CPUDevice, functor::abs<int32>>);
|
| D | cwise_op_ndtri.cc | 20 REGISTER2(UnaryOp, CPU, "Ndtri", functor::ndtri, float, double);
21 REGISTER2(UnaryOp, CPU, "Erfinv", functor::erfinv, float, double);
23 REGISTER2(UnaryOp, GPU, "Ndtri", functor::ndtri, float, double);
24 REGISTER2(UnaryOp, GPU, "Erfinv", functor::erfinv, float, double);
|
| D | cwise_op_sign.cc | 19 REGISTER8(UnaryOp, CPU, "Sign", functor::sign, float, double, int32, int64,
24 REGISTER6(UnaryOp, GPU, "Sign", functor::sign, float, Eigen::half, double,
27 REGISTER2(UnaryOp, GPU, "Sign", functor::sign, complex64, complex128);
38 UnaryOp<CPUDevice, functor::sign<int32>>);
|
| D | cwise_op_reciprocal.cc | 19 REGISTER5(UnaryOp, CPU, "Inv", functor::inverse, float, Eigen::half, double,
22 REGISTER4(UnaryOp, GPU, "Inv", functor::inverse, float, Eigen::half, double,
33 REGISTER6(UnaryOp, CPU, "Reciprocal", functor::inverse, float, Eigen::half,
36 REGISTER4(UnaryOp, GPU, "Reciprocal", functor::inverse, float, Eigen::half,
|
| D | cwise_op_neg_2.cc | 19 REGISTER6(UnaryOp, CPU, "Neg", functor::neg, Eigen::half, float, double,
24 REGISTER3(UnaryOp, GPU, "Neg", functor::neg, Eigen::half, float, double);
26 REGISTER3(UnaryOp, GPU, "Neg", functor::neg, bfloat16, complex64, complex128);
|
| D | cwise_op_neg_1.cc | 19 REGISTER4(UnaryOp, CPU, "Neg", functor::neg, int8, int16, int32, int64);
24 REGISTER3(UnaryOp, GPU, "Neg", functor::neg, int8, int16, int64);
35 UnaryOp<CPUDevice, functor::neg<int32>>);
|
| D | cwise_op_square.cc | 19 REGISTER8(UnaryOp, CPU, "Square", functor::square, float, Eigen::half, double,
24 REGISTER4(UnaryOp, GPU, "Square", functor::square, float, Eigen::half, double,
36 UnaryOp<CPUDevice, functor::square<int32>>);
|
| D | cwise_op_invert.cc | 19 REGISTER8(UnaryOp, CPU, "Invert", functor::invert, int8, int16, int32, int64,
25 REGISTER8(UnaryOp, GPU, "Invert", functor::invert, int8, int16, int32, int64,
28 REGISTER4(UnaryOp, GPU, "Invert", functor::invert, uint8, uint16, uint32,
|
| D | cwise_op_conj.cc | 21 REGISTER2(UnaryOp, CPU, "Conj", functor::conj, complex64, complex128);
32 UnaryOp<GPUDevice, functor::conj<complex64>>);
35 UnaryOp<GPUDevice, functor::conj<complex128>>);
|
| D | cwise_op_rint.cc | 19 REGISTER2(UnaryOp, CPU, "Rint", functor::rint, float, double);
21 REGISTER2(UnaryOp, GPU, "Rint", functor::rint, float, double);
|
| D | cwise_op_acos.cc | 19 REGISTER2(UnaryOp, CPU, "Acos", functor::acos, float, double);
23 REGISTER2(UnaryOp, GPU, "Acos", functor::acos, float, double);
|
| D | cwise_op_atan.cc | 19 REGISTER2(UnaryOp, CPU, "Atan", functor::atan, float, double);
23 REGISTER2(UnaryOp, GPU, "Atan", functor::atan, float, double);
|
| /external/llvm-project/pstl/test/std/algorithms/alg.modifying.operations/alg.partitions/ |
| D | partition.pass.cpp | 69 template <typename BiDirIt, typename Size, typename UnaryOp, typename Generator>
72 Size n, UnaryOp unary_op, Generator generator) in operator ()()
76 template <typename BiDirIt, typename Size, typename UnaryOp, typename Generator>
79 BiDirIt exp_last, Size n, UnaryOp unary_op, Generator generator) in operator ()()
83 template <typename BiDirIt, typename Size, typename UnaryOp, typename Generator>
86 Size n, UnaryOp unary_op, Generator generator)
90 template <typename BiDirIt, typename Size, typename UnaryOp, typename Generator>
93 BiDirIt exp_last, Size n, UnaryOp unary_op, Generator generator)
98 … template <typename Policy, typename BiDirIt, typename Size, typename UnaryOp, typename Generator>
100 …y&& exec, BiDirIt first, BiDirIt last, BiDirIt exp_first, BiDirIt exp_last, Size, UnaryOp unary_op, in operator ()()
[all …]
|
| D | partition_copy.pass.cpp | 28 typename UnaryOp>
31 OutputIterator2 false_first, OutputIterator2, UnaryOp unary_op) in operator ()()
44 …plate <typename InputIterator, typename OutputIterator, typename OutputIterator2, typename UnaryOp>
49 OutputIterator2 false_last, UnaryOp unary_op) in operator ()()
52 …plate <typename InputIterator, typename OutputIterator, typename OutputIterator2, typename UnaryOp>
57 OutputIterator2 false_last, UnaryOp unary_op) in operator ()()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorExpr.h | 80 template<typename UnaryOp, typename XprType>
81 struct traits<TensorCwiseUnaryOp<UnaryOp, XprType> >
86 typedef typename result_of<UnaryOp(typename XprType::Scalar)>::type Scalar;
94 template<typename UnaryOp, typename XprType>
95 struct eval<TensorCwiseUnaryOp<UnaryOp, XprType>, Eigen::Dense>
97 typedef const TensorCwiseUnaryOp<UnaryOp, XprType>& type;
100 template<typename UnaryOp, typename XprType>
101 struct nested<TensorCwiseUnaryOp<UnaryOp, XprType>, 1, typename eval<TensorCwiseUnaryOp<UnaryOp, Xp…
103 typedef TensorCwiseUnaryOp<UnaryOp, XprType> type;
110 template<typename UnaryOp, typename XprType>
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | numeric_op.h | 30 class UnaryOp : public OpKernel {
32 explicit UnaryOp(OpKernelConstruction* context) : OpKernel(context) { in UnaryOp() function
52 class UnaryElementWiseOp : public UnaryOp<T> {
54 using UnaryOp<T>::UnaryOp;
|
| /external/llvm-project/pstl/test/std/numerics/numeric.ops/ |
| D | transform_scan.pass.cpp | 38 …<typename Policy, typename InputIterator, typename OutputIterator, typename Size, typename UnaryOp,
42 …Iterator out_last, OutputIterator expected_first, OutputIterator, Size n, UnaryOp unary_op, T init, in operator ()()
66 …<typename Policy, typename InputIterator, typename OutputIterator, typename Size, typename UnaryOp,
70 Size, UnaryOp, T, BinaryOp, T) in operator ()()
105 template <typename In, typename Out, typename UnaryOp, typename BinaryOp>
107 test(UnaryOp unary_op, Out init, BinaryOp binary_op, Out trash) in test()
144 template <typename In, typename Out, typename UnaryOp, typename BinaryOp>
146 test_matrix(UnaryOp unary_op, Out init, BinaryOp binary_op, Out trash) in test_matrix()
|