HomeSort by relevance Sort by last modified time
    Searched defs:TensorEvaluator (Results 26 - 30 of 30) sorted by null

12

  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
TensorContractionCuda.h     [all...]
TensorConvolution.h 291 struct TensorEvaluator<const TensorConvolutionOp<Indices, InputArgType, KernelArgType>, Device>
295 static const int NumDims = internal::array_size<typename TensorEvaluator<InputArgType, Device>::Dimensions>::value;
306 IsAligned = TensorEvaluator<InputArgType, Device>::IsAligned & TensorEvaluator<KernelArgType, Device>::IsAligned,
307 PacketAccess = TensorEvaluator<InputArgType, Device>::PacketAccess & TensorEvaluator<KernelArgType, Device>::PacketAccess,
308 Layout = TensorEvaluator<InputArgType, Device>::Layout,
313 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
316 EIGEN_STATIC_ASSERT((static_cast<int>(TensorEvaluator<InputArgType, Device>::Layout) == static_cast<int>(TensorEvaluator<KernelArgType, Device>::Layout)), YOU_MADE_A_PROGRAMMING_MISTAKE)
    [all...]
TensorImagePatch.h 151 struct TensorEvaluator<const TensorImagePatchOp<Rows, Cols, ArgType>, Device>
155 static const int NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
159 typedef TensorEvaluator<const TensorImagePatchOp<Rows, Cols, ArgType>,
161 typedef TensorEvaluator<ArgType, Device> Impl;
168 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
169 Layout = TensorEvaluator<ArgType, Device>::Layout,
174 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
181 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
423 const TensorEvaluator<ArgType, Device>& impl() const { return m_impl; }
440 // TensorEvaluator<ArgType, Device>::CoordAccess is false
    [all...]
TensorContractionThreadPool.h 60 struct TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgType>, ThreadPoolDevice> :
61 public TensorContractionEvaluatorBase<TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgType>, ThreadPoolDevice> > {
65 typedef TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgType>, Device> Self;
75 Layout = TensorEvaluator<LeftArgType, Device>::Layout,
88 internal::array_size<typename TensorEvaluator<EvalLeftArgType, Device>::Dimensions>::value;
90 internal::array_size<typename TensorEvaluator<EvalRightArgType, Device>::Dimensions>::value;
109 typedef TensorEvaluator<EvalLeftArgType, Device> LeftEvaluator;
110 typedef TensorEvaluator<EvalRightArgType, Device> RightEvaluator;
112 TensorEvaluator(const XprType& op, const Device& device) :
126 typedef TensorEvaluator<EvalLeftArgType, Device> LeftEvaluator
    [all...]
TensorVolumePatch.h 166 struct TensorEvaluator<const TensorVolumePatchOp<Planes, Rows, Cols, ArgType>, Device>
170 static const int NumInputDims = internal::array_size<typename TensorEvaluator<ArgType, Device>::Dimensions>::value;
180 PacketAccess = TensorEvaluator<ArgType, Device>::PacketAccess,
182 Layout = TensorEvaluator<ArgType, Device>::Layout,
187 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device)
194 const typename TensorEvaluator<ArgType, Device>::Dimensions& input_dims = m_impl.dimensions();
507 const TensorEvaluator<ArgType, Device>& impl() const { return m_impl; }
602 TensorEvaluator<ArgType, Device> m_impl;

Completed in 1331 milliseconds

12