#include <TensorConvolution.h>
Public Types | |
enum | { IsAligned = TensorEvaluator<InputArgType, Device>::IsAligned & TensorEvaluator<KernelArgType, Device>::IsAligned, PacketAccess = TensorEvaluator<InputArgType, Device>::PacketAccess & TensorEvaluator<KernelArgType, Device>::PacketAccess, Layout = TensorEvaluator<InputArgType, Device>::Layout, CoordAccess = false, RawAccess = false } |
typedef XprType::CoeffReturnType | CoeffReturnType |
typedef DSizes< Index, NumDims > | Dimensions |
typedef XprType::Index | Index |
typedef PacketType< CoeffReturnType, Device >::type | PacketReturnType |
typedef XprType::Scalar | Scalar |
typedef TensorConvolutionOp< Indices, InputArgType, KernelArgType > | XprType |
Public Member Functions | |
EIGEN_DEVICE_FUNC 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 Scalar * | data () const |
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Dimensions & | dimensions () const |
EIGEN_DEVICE_FUNC 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_DEVICE_FUNC 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 = internal::unpacket_traits<PacketReturnType>::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 & | 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 291 of file TensorConvolution.h.
typedef XprType::CoeffReturnType Eigen::TensorEvaluator< const TensorConvolutionOp< Indices, InputArgType, KernelArgType >, Device >::CoeffReturnType |
Definition at line 301 of file TensorConvolution.h.
typedef DSizes<Index, NumDims> Eigen::TensorEvaluator< const TensorConvolutionOp< Indices, InputArgType, KernelArgType >, Device >::Dimensions |
Definition at line 298 of file TensorConvolution.h.
typedef XprType::Index Eigen::TensorEvaluator< const TensorConvolutionOp< Indices, InputArgType, KernelArgType >, Device >::Index |
Definition at line 297 of file TensorConvolution.h.
typedef PacketType<CoeffReturnType, Device>::type Eigen::TensorEvaluator< const TensorConvolutionOp< Indices, InputArgType, KernelArgType >, Device >::PacketReturnType |
Definition at line 302 of file TensorConvolution.h.
typedef XprType::Scalar Eigen::TensorEvaluator< const TensorConvolutionOp< Indices, InputArgType, KernelArgType >, Device >::Scalar |
Definition at line 300 of file TensorConvolution.h.
typedef TensorConvolutionOp<Indices, InputArgType, KernelArgType> Eigen::TensorEvaluator< const TensorConvolutionOp< Indices, InputArgType, KernelArgType >, Device >::XprType |
Definition at line 293 of file TensorConvolution.h.
anonymous enum |
Enumerator | |
---|---|
IsAligned | |
PacketAccess | |
Layout | |
CoordAccess | |
RawAccess |
Definition at line 305 of file TensorConvolution.h.
|
inline |
Definition at line 313 of file TensorConvolution.h.
|
inline |
Definition at line 382 of file TensorConvolution.h.
|
inline |
Definition at line 399 of file TensorConvolution.h.
|
inlineprivate |
Definition at line 490 of file TensorConvolution.h.
|
inlineprivate |
Definition at line 503 of file TensorConvolution.h.
|
inline |
Definition at line 452 of file TensorConvolution.h.
|
inline |
Definition at line 468 of file TensorConvolution.h.
|
inline |
Definition at line 375 of file TensorConvolution.h.
|
inline |
Definition at line 377 of file TensorConvolution.h.
|
inline |
Definition at line 391 of file TensorConvolution.h.
|
inlineprivate |
Definition at line 471 of file TensorConvolution.h.
|
inline |
Definition at line 407 of file TensorConvolution.h.
|
inlineprivate |
Definition at line 515 of file TensorConvolution.h.
|
private |
Definition at line 547 of file TensorConvolution.h.
|
private |
Definition at line 542 of file TensorConvolution.h.
|
private |
Definition at line 538 of file TensorConvolution.h.
|
private |
Definition at line 540 of file TensorConvolution.h.
|
private |
Definition at line 535 of file TensorConvolution.h.
|
private |
Definition at line 545 of file TensorConvolution.h.
|
private |
Definition at line 544 of file TensorConvolution.h.
|
private |
Definition at line 541 of file TensorConvolution.h.
|
private |
Definition at line 539 of file TensorConvolution.h.
|
private |
Definition at line 546 of file TensorConvolution.h.
|
private |
Definition at line 536 of file TensorConvolution.h.
|
static |
Definition at line 295 of file TensorConvolution.h.
|
static |
Definition at line 296 of file TensorConvolution.h.
|
static |
Definition at line 303 of file TensorConvolution.h.