Home
last modified time | relevance | path

Searched refs:UnaryOp (Results 1 – 25 of 119) sorted by relevance

12345

/external/eigen/Eigen/src/Core/
DCwiseUnaryOp.h17 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/
DSparseCwiseUnaryOp.h17 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/
Dcwise_op_abs.cc19 REGISTER5(UnaryOp, CPU, "Abs", functor::abs, float, Eigen::half, double, int32,
21 REGISTER2(UnaryOp, CPU, "ComplexAbs", functor::abs, complex64, complex128);
24 REGISTER4(UnaryOp, GPU, "Abs", functor::abs, float, Eigen::half, double, int64);
25 REGISTER2(UnaryOp, GPU, "ComplexAbs", functor::abs, complex64, complex128);
35 UnaryOp<CPUDevice, functor::abs<int32>>);
39 REGISTER3(UnaryOp, SYCL, "Abs", functor::abs, float, double, int64);
45 UnaryOp<CPUDevice, functor::abs<int32>>);
Dcwise_op_square.cc19 REGISTER8(UnaryOp, CPU, "Square", functor::square, float, Eigen::half, double,
23 REGISTER4(UnaryOp, GPU, "Square", functor::square, float, Eigen::half, double,
34 UnaryOp<CPUDevice, functor::square<int32>>);
38 REGISTER3(UnaryOp, SYCL, "Square", functor::square, float, double, int64);
44 UnaryOp<CPUDevice, functor::square<int32>>);
Dcwise_op_neg.cc19 REGISTER8(UnaryOp, CPU, "Neg", functor::neg, float, Eigen::half, double, int32,
23 REGISTER3(UnaryOp, SYCL, "Neg", functor::neg, float, double, int64);
29 UnaryOp<CPUDevice, functor::neg<int32>>);
33 REGISTER6(UnaryOp, GPU, "Neg", functor::neg, float, Eigen::half, double, int64,
44 UnaryOp<CPUDevice, functor::neg<int32>>);
Dcwise_op_sign.cc19 REGISTER7(UnaryOp, CPU, "Sign", functor::sign, float, double, int32, int64,
22 REGISTER6(UnaryOp, GPU, "Sign", functor::sign, float, Eigen::half, double,
33 UnaryOp<CPUDevice, functor::sign<int32>>);
37 REGISTER3(UnaryOp, SYCL, "Sign", functor::sign, float, double, int64);
43 UnaryOp<CPUDevice, functor::sign<int32>>);
Dcwise_op_reciprocal.cc19 REGISTER5(UnaryOp, CPU, "Inv", functor::inverse, float, Eigen::half, double,
22 REGISTER4(UnaryOp, GPU, "Inv", functor::inverse, float, Eigen::half, double,
33 REGISTER5(UnaryOp, CPU, "Reciprocal", functor::inverse, float, Eigen::half,
36 REGISTER4(UnaryOp, GPU, "Reciprocal", functor::inverse, float, Eigen::half,
40 REGISTER(UnaryOp, SYCL, "Reciprocal", functor::inverse, float);
Dcwise_op_bessel.cc19 REGISTER3(UnaryOp, CPU, "BesselI0e", functor::bessel_i0e, Eigen::half, float,
21 REGISTER3(UnaryOp, CPU, "BesselI1e", functor::bessel_i1e, Eigen::half, float,
24 REGISTER3(UnaryOp, GPU, "BesselI0e", functor::bessel_i0e, Eigen::half, float,
26 REGISTER3(UnaryOp, GPU, "BesselI1e", functor::bessel_i1e, Eigen::half, float,
Dcwise_op_asin.cc19 REGISTER2(UnaryOp, CPU, "Asin", functor::asin, float, double);
22 REGISTER2(UnaryOp, GPU, "Asin", functor::asin, float, double);
26 REGISTER2(UnaryOp, SYCL, "Asin", functor::asin, float, double);
Dcwise_op_atan.cc19 REGISTER2(UnaryOp, CPU, "Atan", functor::atan, float, double);
22 REGISTER2(UnaryOp, GPU, "Atan", functor::atan, float, double);
26 REGISTER2(UnaryOp, SYCL, "Atan", functor::atan, float, double);
Dcwise_op_acos.cc19 REGISTER2(UnaryOp, CPU, "Acos", functor::acos, float, double);
22 REGISTER2(UnaryOp, GPU, "Acos", functor::acos, float, double);
26 REGISTER2(UnaryOp, SYCL, "Acos", functor::acos, float, double);
Dcwise_op_acosh.cc20 REGISTER4(UnaryOp, CPU, "Acosh", functor::acosh, float, double, complex64,
24 REGISTER2(UnaryOp, SYCL, "Acosh", functor::acosh, float, double);
28 REGISTER2(UnaryOp, GPU, "Acosh", functor::acosh, float, double);
Dcwise_op_atanh.cc20 REGISTER4(UnaryOp, CPU, "Atanh", functor::atanh, float, double, complex64,
24 REGISTER2(UnaryOp, SYCL, "Atanh", functor::atanh, float, double);
28 REGISTER2(UnaryOp, GPU, "Atanh", functor::atanh, float, double);
Dcwise_op_asinh.cc20 REGISTER4(UnaryOp, CPU, "Asinh", functor::asinh, float, double, complex64,
24 REGISTER2(UnaryOp, SYCL, "Asinh", functor::asinh, float, double);
28 REGISTER2(UnaryOp, GPU, "Asinh", functor::asinh, float, double);
Dcwise_op_isinf.cc19 REGISTER3(UnaryOp, CPU, "IsInf", functor::isinf, float, Eigen::half, double);
22 REGISTER3(UnaryOp, GPU, "IsInf", functor::isinf, float, Eigen::half, double);
26 REGISTER2(UnaryOp, SYCL, "IsInf", functor::isinf, float, double);
Dcwise_op_ceil.cc19 REGISTER3(UnaryOp, CPU, "Ceil", functor::ceil, float, Eigen::half, double);
22 REGISTER3(UnaryOp, GPU, "Ceil", functor::ceil, float, Eigen::half, double);
26 REGISTER2(UnaryOp, SYCL, "Ceil", functor::ceil, float, double);
Dcwise_op_floor.cc19 REGISTER3(UnaryOp, CPU, "Floor", functor::floor, float, Eigen::half, double);
22 REGISTER3(UnaryOp, GPU, "Floor", functor::floor, float, Eigen::half, double);
25 REGISTER2(UnaryOp, SYCL, "Floor", functor::floor, float, double);
Dcwise_op_isfinite.cc19 REGISTER3(UnaryOp, CPU, "IsFinite", functor::isfinite, float, Eigen::half,
23 REGISTER3(UnaryOp, GPU, "IsFinite", functor::isfinite, float, Eigen::half,
28 REGISTER2(UnaryOp, SYCL, "IsFinite", functor::isfinite, float, double);
Dcwise_op_isnan.cc19 REGISTER4(UnaryOp, CPU, "IsNan", functor::isnan, float, Eigen::half, double,
23 REGISTER3(UnaryOp, GPU, "IsNan", functor::isnan, float, Eigen::half, double);
27 REGISTER2(UnaryOp, SYCL, "IsNan", functor::isnan, float, double);
Dcwise_op_expm1.cc19 REGISTER5(UnaryOp, CPU, "Expm1", functor::expm1, float, Eigen::half, double,
22 REGISTER3(UnaryOp, GPU, "Expm1", functor::expm1, float, Eigen::half, double);
25 REGISTER2(UnaryOp, SYCL, "Expm1", functor::expm1, float, double);
Dcwise_op_tan.cc19 REGISTER5(UnaryOp, CPU, "Tan", functor::tan, Eigen::half, float, double,
23 REGISTER3(UnaryOp, GPU, "Tan", functor::tan, Eigen::half, float, double);
27 REGISTER2(UnaryOp, SYCL, "Tan", functor::tan, float, double);
Dcwise_op_log1p.cc19 REGISTER5(UnaryOp, CPU, "Log1p", functor::log1p, float, Eigen::half, double,
23 REGISTER3(UnaryOp, GPU, "Log1p", functor::log1p, float, Eigen::half, double);
27 REGISTER2(UnaryOp, SYCL, "Log1p", functor::log1p, float, double);
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorExpr.h80 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/compiler/jit/
Dxla_fusion_optimizer_test.cc53 Node* b = ops::UnaryOp("Relu", a, builder.opts().WithName("B")); in TEST_F()
54 Node* c = ops::UnaryOp("Relu", b, builder.opts().WithName("C")); in TEST_F()
56 ops::UnaryOp("UncompilableUnary", c, builder.opts().WithName("D")); in TEST_F()
57 Node* e = ops::UnaryOp("Relu", d, builder.opts().WithName("E")); in TEST_F()
58 ops::UnaryOp("Relu", e, builder.opts().WithName("F")); in TEST_F()
89 ops::UnaryOp("Abs", c, builder.opts().WithName("E")); in TEST_F()
120 Node* e = ops::UnaryOp("Abs", c, builder.opts().WithName("E")); in TEST_F()
121 ops::UnaryOp("Cos", e, in TEST_F()
145 ops::UnaryOp("UncompilableUnary", a, builder.opts().WithName("B")); in TEST_F()
168 Node* b = ops::UnaryOp("Relu", a, builder.opts().WithName("B")); in TEST_F()
/external/tensorflow/tensorflow/core/framework/
Dnumeric_op.h30 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;

12345