#include <TensorConvolution.h>
Public Member Functions | |
IndexMapper (const InputDims &input_dims, const array< Index, NumKernelDims > &kernel_dims, const array< Index, NumKernelDims > &indices) | |
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index | mapGpuInputKernelToTensorInputOffset (Index i) const |
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index | mapGpuInputKernelToTensorInputOffset (Index i, Index j) const |
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index | mapGpuInputKernelToTensorInputOffset (Index i, Index j, Index k) const |
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index | mapGpuInputPlaneToTensorInputOffset (Index p) const |
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index | mapGpuOutputKernelToTensorOutputOffset (Index i) const |
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index | mapGpuOutputKernelToTensorOutputOffset (Index i, Index j) const |
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index | mapGpuOutputKernelToTensorOutputOffset (Index i, Index j, Index k) const |
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index | mapGpuOutputPlaneToTensorOutputOffset (Index p) const |
Private Attributes | |
array< Index, NumDims > | m_gpuInputStrides |
array< Index, NumDims > | m_gpuOutputStrides |
array< Index, NumDims > | m_inputStrides |
array< Index, NumDims > | m_outputStrides |
Static Private Attributes | |
static const int | NumDims = internal::array_size<InputDims>::value |
Definition at line 25 of file TensorConvolution.h.
|
inline |
Definition at line 27 of file TensorConvolution.h.
|
inline |
Definition at line 164 of file TensorConvolution.h.
|
inline |
Definition at line 178 of file TensorConvolution.h.
|
inline |
Definition at line 192 of file TensorConvolution.h.
|
inline |
Definition at line 116 of file TensorConvolution.h.
|
inline |
Definition at line 171 of file TensorConvolution.h.
|
inline |
Definition at line 185 of file TensorConvolution.h.
|
inline |
Definition at line 200 of file TensorConvolution.h.
|
inline |
Definition at line 140 of file TensorConvolution.h.
|
private |
Definition at line 212 of file TensorConvolution.h.
|
private |
Definition at line 213 of file TensorConvolution.h.
|
private |
Definition at line 210 of file TensorConvolution.h.
|
private |
Definition at line 211 of file TensorConvolution.h.
|
staticprivate |
Definition at line 209 of file TensorConvolution.h.