Searched refs:TensorCustomUnaryOp (Results 1 – 4 of 4) sorted by relevance
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
D | TensorCustomOp.h | 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>, ReadOn… 55 typedef typename internal::traits<TensorCustomUnaryOp>::Scalar Scalar; 58 typedef typename internal::nested<TensorCustomUnaryOp>::type Nested; 59 typedef typename internal::traits<TensorCustomUnaryOp>::StorageKind StorageKind; 60 typedef typename internal::traits<TensorCustomUnaryOp>::Index Index; [all …]
|
D | TensorForwardDeclarations.h | 62 template<typename CustomUnaryFunc, typename XprType> class TensorCustomUnaryOp; variable
|
D | TensorBase.h | 803 …const TensorCustomUnaryOp<const CustomUnaryFunc, const Derived> customOp(const CustomUnaryFunc& op… in customOp() 804 return TensorCustomUnaryOp<const CustomUnaryFunc, const Derived>(derived(), op); in customOp()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | eigen_attention.h | 275 EIGEN_ALWAYS_INLINE static const TensorCustomUnaryOp<
|