Searched refs:TensorCustomUnaryOp (Results 1 - 5 of 5) sorted by relevance
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
H A D | TensorCustomOp.h | 15 /** \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 D | TensorForwardDeclarations.h | 62 template<typename CustomUnaryFunc, typename XprType> class TensorCustomUnaryOp;
|
H A D | TensorBase.h | 803 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 D | eigen_softmax.h | 85 EIGEN_ALWAYS_INLINE static const TensorCustomUnaryOp<const SoftmaxOp,
|
H A D | eigen_attention.h | 233 EIGEN_ALWAYS_INLINE static const TensorCustomUnaryOp<
|
Completed in 278 milliseconds