Searched refs:TensorCustomUnaryOp (Results 1 - 5 of 5) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorCustomOp.h15 /** \class TensorCustomUnaryOp
24 struct traits<TensorCustomUnaryOp<CustomUnaryFunc, XprType> >
36 struct eval<TensorCustomUnaryOp<CustomUnaryFunc, XprType>, Eigen::Dense>
38 typedef const TensorCustomUnaryOp<CustomUnaryFunc, XprType>& type;
42 struct nested<TensorCustomUnaryOp<CustomUnaryFunc, XprType> >
44 typedef TensorCustomUnaryOp<CustomUnaryFunc, XprType> type;
52 class TensorCustomUnaryOp : public TensorBase<TensorCustomUnaryOp<CustomUnaryFunc, XprType>, ReadOnlyAccessors> class in namespace:Eigen
55 typedef typename internal::traits<TensorCustomUnaryOp>::Scalar Scalar;
58 typedef typename internal::nested<TensorCustomUnaryOp>
62 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorCustomUnaryOp(const XprType& expr, const CustomUnaryFunc& func) function in class:Eigen::TensorCustomUnaryOp
[all...]
H A DTensorForwardDeclarations.h62 template<typename CustomUnaryFunc, typename XprType> class TensorCustomUnaryOp;
H A DTensorBase.h803 const TensorCustomUnaryOp<const CustomUnaryFunc, const Derived> customOp(const CustomUnaryFunc& op) const {
804 return TensorCustomUnaryOp<const CustomUnaryFunc, const Derived>(derived(), op);
/external/tensorflow/tensorflow/core/kernels/
H A Deigen_softmax.h85 EIGEN_ALWAYS_INLINE static const TensorCustomUnaryOp<const SoftmaxOp,
H A Deigen_attention.h233 EIGEN_ALWAYS_INLINE static const TensorCustomUnaryOp<

Completed in 278 milliseconds