Home
last modified time | relevance | path

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

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorCustomOp.h24 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 …]
DTensorForwardDeclarations.h62 template<typename CustomUnaryFunc, typename XprType> class TensorCustomUnaryOp; variable
DTensorBase.h803 …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/
Deigen_softmax.h85 EIGEN_ALWAYS_INLINE static const TensorCustomUnaryOp<const SoftmaxOp,
Deigen_attention.h250 EIGEN_ALWAYS_INLINE static const TensorCustomUnaryOp<