#include <TensorConvolution.h>
Public Types | |
enum | { IsAligned = int(TensorEvaluator<InputArgType, Device>::IsAligned) & int(TensorEvaluator<KernelArgType, Device>::IsAligned), PacketAccess = int(TensorEvaluator<InputArgType, Device>::PacketAccess) & int(TensorEvaluator<KernelArgType, Device>::PacketAccess), BlockAccess = false, PreferBlockAccess = false, Layout = TensorEvaluator<InputArgType, Device>::Layout, CoordAccess = false, RawAccess = false } |
typedef XprType::CoeffReturnType | CoeffReturnType |
typedef DSizes< Index, NumDims > | Dimensions |
typedef Storage::Type | EvaluatorPointerType |
typedef XprType::Index | Index |
typedef PacketType< CoeffReturnType, Device >::type | PacketReturnType |
typedef XprType::Scalar | Scalar |
typedef StorageMemory< Scalar, Device > | Storage |
typedef internal::TensorBlockNotImplemented | TensorBlock |
typedef TensorConvolutionOp< Indices, InputArgType, KernelArgType > | XprType |
Public Member Functions | |
EIGEN_STRONG_INLINE void | cleanup () |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE CoeffReturnType | coeff (Index index) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorOpCost | costPerCoeff (bool vectorized) const |
EIGEN_DEVICE_FUNC EvaluatorPointerType | data () const |
EIGEN_DEVICE_FUNC const EIGEN_STRONG_INLINE Dimensions & | dimensions () const |
EIGEN_STRONG_INLINE bool | evalSubExprsIfNeeded (Scalar *) |
void | evalTo (typename XprType::Scalar *buffer) |
template<int LoadMode> | |
EIGEN_DEVICE_FUNC PacketReturnType | packet (const Index index) const |
EIGEN_STRONG_INLINE | TensorEvaluator (const XprType &op, const Device &device) |
Static Public Attributes | |
static const int | NumDims = internal::array_size<typename TensorEvaluator<InputArgType, Device>::Dimensions>::value |
static const int | NumKernelDims = internal::array_size<Indices>::value |
static const int | PacketSize = PacketType<CoeffReturnType, Device>::size |
Private Member Functions | |
EIGEN_DEVICE_FUNC void | convolve (Index firstIndex, Index firstKernel, int DimIndex, CoeffReturnType &accum) const |
template<typename Packet > | |
EIGEN_DEVICE_FUNC void | convolvePacket (Index firstIndex, Index firstKernel, int DimIndex, Packet &accum) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Index | firstInput (Index index) const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void | preloadKernel () |
Private Attributes | |
const Device EIGEN_DEVICE_REF | m_device |
Dimensions | m_dimensions |
array< Index, NumKernelDims > | m_indexStride |
TensorEvaluator< InputArgType, Device > | m_inputImpl |
array< Index, NumDims > | m_inputStride |
const Scalar * | m_kernel |
KernelArgType | m_kernelArg |
TensorEvaluator< KernelArgType, Device > | m_kernelImpl |
array< Index, NumKernelDims > | m_kernelStride |
bool | m_local_kernel |
array< Index, NumDims > | m_outputStride |
Definition at line 293 of file TensorConvolution.h.
typedef XprType::CoeffReturnType Eigen::TensorEvaluator< const TensorConvolutionOp< Indices, InputArgType, KernelArgType >, Device >::CoeffReturnType |
Definition at line 303 of file TensorConvolution.h.
typedef DSizes<Index, NumDims> Eigen::TensorEvaluator< const TensorConvolutionOp< Indices, InputArgType, KernelArgType >, Device >::Dimensions |
Definition at line 300 of file TensorConvolution.h.
typedef Storage::Type Eigen::TensorEvaluator< const TensorConvolutionOp< Indices, InputArgType, KernelArgType >, Device >::EvaluatorPointerType |
Definition at line 307 of file TensorConvolution.h.
typedef XprType::Index Eigen::TensorEvaluator< const TensorConvolutionOp< Indices, InputArgType, KernelArgType >, Device >::Index |
Definition at line 299 of file TensorConvolution.h.
typedef PacketType<CoeffReturnType, Device>::type Eigen::TensorEvaluator< const TensorConvolutionOp< Indices, InputArgType, KernelArgType >, Device >::PacketReturnType |
Definition at line 304 of file TensorConvolution.h.
typedef XprType::Scalar Eigen::TensorEvaluator< const TensorConvolutionOp< Indices, InputArgType, KernelArgType >, Device >::Scalar |
Definition at line 302 of file TensorConvolution.h.
typedef StorageMemory<Scalar, Device> Eigen::TensorEvaluator< const TensorConvolutionOp< Indices, InputArgType, KernelArgType >, Device >::Storage |
Definition at line 306 of file TensorConvolution.h.
typedef internal::TensorBlockNotImplemented Eigen::TensorEvaluator< const TensorConvolutionOp< Indices, InputArgType, KernelArgType >, Device >::TensorBlock |
Definition at line 320 of file TensorConvolution.h.
typedef TensorConvolutionOp<Indices, InputArgType, KernelArgType> Eigen::TensorEvaluator< const TensorConvolutionOp< Indices, InputArgType, KernelArgType >, Device >::XprType |
Definition at line 295 of file TensorConvolution.h.
anonymous enum |
Enumerator | |
---|---|
IsAligned | |
PacketAccess | |
BlockAccess | |
PreferBlockAccess | |
Layout | |
CoordAccess | |
RawAccess |
Definition at line 309 of file TensorConvolution.h.
|
inline |
Definition at line 323 of file TensorConvolution.h.
|
inline |
Definition at line 392 of file TensorConvolution.h.
|
inline |
Definition at line 409 of file TensorConvolution.h.
|
inlineprivate |
Definition at line 500 of file TensorConvolution.h.
|
inlineprivate |
Definition at line 513 of file TensorConvolution.h.
|
inline |
Definition at line 462 of file TensorConvolution.h.
|
inline |
Definition at line 478 of file TensorConvolution.h.
|
inline |
Definition at line 385 of file TensorConvolution.h.
|
inline |
Definition at line 387 of file TensorConvolution.h.
|
inline |
Definition at line 401 of file TensorConvolution.h.
|
inlineprivate |
Definition at line 481 of file TensorConvolution.h.
|
inline |
Definition at line 417 of file TensorConvolution.h.
|
inlineprivate |
Definition at line 525 of file TensorConvolution.h.
|
private |
Definition at line 557 of file TensorConvolution.h.
|
private |
Definition at line 552 of file TensorConvolution.h.
|
private |
Definition at line 548 of file TensorConvolution.h.
|
private |
Definition at line 550 of file TensorConvolution.h.
|
private |
Definition at line 545 of file TensorConvolution.h.
|
private |
Definition at line 555 of file TensorConvolution.h.
|
private |
Definition at line 554 of file TensorConvolution.h.
|
private |
Definition at line 551 of file TensorConvolution.h.
|
private |
Definition at line 549 of file TensorConvolution.h.
|
private |
Definition at line 556 of file TensorConvolution.h.
|
private |
Definition at line 546 of file TensorConvolution.h.
|
static |
Definition at line 297 of file TensorConvolution.h.
|
static |
Definition at line 298 of file TensorConvolution.h.
|
static |
Definition at line 305 of file TensorConvolution.h.