HomeSort by relevance Sort by last modified time
    Searched refs:TensorCustomUnaryOp (Results 1 - 5 of 5) sorted by null

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
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>
55 typedef typename internal::traits<TensorCustomUnaryOp>::Scalar Scalar;
58 typedef typename internal::nested<TensorCustomUnaryOp>::type Nested
    [all...]
TensorForwardDeclarations.h 62 template<typename CustomUnaryFunc, typename XprType> class TensorCustomUnaryOp;
TensorBase.h     [all...]
  /external/tensorflow/tensorflow/core/kernels/
eigen_softmax.h 85 EIGEN_ALWAYS_INLINE static const TensorCustomUnaryOp<const SoftmaxOp,
eigen_attention.h 233 EIGEN_ALWAYS_INLINE static const TensorCustomUnaryOp<

Completed in 244 milliseconds