Public Member Functions | Private Attributes | Static Private Attributes | List of all members
Eigen::internal::IndexMapper< Index, InputDims, NumKernelDims, Layout > Class Template Reference

#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, NumDimsm_gpuInputStrides
 
array< Index, NumDimsm_gpuOutputStrides
 
array< Index, NumDimsm_inputStrides
 
array< Index, NumDimsm_outputStrides
 

Static Private Attributes

static const int NumDims = internal::array_size<InputDims>::value
 

Detailed Description

template<typename Index, typename InputDims, int NumKernelDims, int Layout>
class Eigen::internal::IndexMapper< Index, InputDims, NumKernelDims, Layout >

Definition at line 25 of file TensorConvolution.h.

Constructor & Destructor Documentation

◆ IndexMapper()

template<typename Index , typename InputDims , int NumKernelDims, int Layout>
Eigen::internal::IndexMapper< Index, InputDims, NumKernelDims, Layout >::IndexMapper ( const InputDims &  input_dims,
const array< Index, NumKernelDims > &  kernel_dims,
const array< Index, NumKernelDims > &  indices 
)
inline

Definition at line 27 of file TensorConvolution.h.

Member Function Documentation

◆ mapGpuInputKernelToTensorInputOffset() [1/3]

template<typename Index , typename InputDims , int NumKernelDims, int Layout>
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index Eigen::internal::IndexMapper< Index, InputDims, NumKernelDims, Layout >::mapGpuInputKernelToTensorInputOffset ( Index  i) const
inline

Definition at line 164 of file TensorConvolution.h.

◆ mapGpuInputKernelToTensorInputOffset() [2/3]

template<typename Index , typename InputDims , int NumKernelDims, int Layout>
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index Eigen::internal::IndexMapper< Index, InputDims, NumKernelDims, Layout >::mapGpuInputKernelToTensorInputOffset ( Index  i,
Index  j 
) const
inline

Definition at line 178 of file TensorConvolution.h.

◆ mapGpuInputKernelToTensorInputOffset() [3/3]

template<typename Index , typename InputDims , int NumKernelDims, int Layout>
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index Eigen::internal::IndexMapper< Index, InputDims, NumKernelDims, Layout >::mapGpuInputKernelToTensorInputOffset ( Index  i,
Index  j,
Index  k 
) const
inline

Definition at line 192 of file TensorConvolution.h.

◆ mapGpuInputPlaneToTensorInputOffset()

template<typename Index , typename InputDims , int NumKernelDims, int Layout>
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index Eigen::internal::IndexMapper< Index, InputDims, NumKernelDims, Layout >::mapGpuInputPlaneToTensorInputOffset ( Index  p) const
inline

Definition at line 116 of file TensorConvolution.h.

◆ mapGpuOutputKernelToTensorOutputOffset() [1/3]

template<typename Index , typename InputDims , int NumKernelDims, int Layout>
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index Eigen::internal::IndexMapper< Index, InputDims, NumKernelDims, Layout >::mapGpuOutputKernelToTensorOutputOffset ( Index  i) const
inline

Definition at line 171 of file TensorConvolution.h.

◆ mapGpuOutputKernelToTensorOutputOffset() [2/3]

template<typename Index , typename InputDims , int NumKernelDims, int Layout>
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index Eigen::internal::IndexMapper< Index, InputDims, NumKernelDims, Layout >::mapGpuOutputKernelToTensorOutputOffset ( Index  i,
Index  j 
) const
inline

Definition at line 185 of file TensorConvolution.h.

◆ mapGpuOutputKernelToTensorOutputOffset() [3/3]

template<typename Index , typename InputDims , int NumKernelDims, int Layout>
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index Eigen::internal::IndexMapper< Index, InputDims, NumKernelDims, Layout >::mapGpuOutputKernelToTensorOutputOffset ( Index  i,
Index  j,
Index  k 
) const
inline

Definition at line 200 of file TensorConvolution.h.

◆ mapGpuOutputPlaneToTensorOutputOffset()

template<typename Index , typename InputDims , int NumKernelDims, int Layout>
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index Eigen::internal::IndexMapper< Index, InputDims, NumKernelDims, Layout >::mapGpuOutputPlaneToTensorOutputOffset ( Index  p) const
inline

Definition at line 140 of file TensorConvolution.h.

Member Data Documentation

◆ m_gpuInputStrides

template<typename Index , typename InputDims , int NumKernelDims, int Layout>
array<Index, NumDims> Eigen::internal::IndexMapper< Index, InputDims, NumKernelDims, Layout >::m_gpuInputStrides
private

Definition at line 212 of file TensorConvolution.h.

◆ m_gpuOutputStrides

template<typename Index , typename InputDims , int NumKernelDims, int Layout>
array<Index, NumDims> Eigen::internal::IndexMapper< Index, InputDims, NumKernelDims, Layout >::m_gpuOutputStrides
private

Definition at line 213 of file TensorConvolution.h.

◆ m_inputStrides

template<typename Index , typename InputDims , int NumKernelDims, int Layout>
array<Index, NumDims> Eigen::internal::IndexMapper< Index, InputDims, NumKernelDims, Layout >::m_inputStrides
private

Definition at line 210 of file TensorConvolution.h.

◆ m_outputStrides

template<typename Index , typename InputDims , int NumKernelDims, int Layout>
array<Index, NumDims> Eigen::internal::IndexMapper< Index, InputDims, NumKernelDims, Layout >::m_outputStrides
private

Definition at line 211 of file TensorConvolution.h.

◆ NumDims

template<typename Index , typename InputDims , int NumKernelDims, int Layout>
const int Eigen::internal::IndexMapper< Index, InputDims, NumKernelDims, Layout >::NumDims = internal::array_size<InputDims>::value
staticprivate

Definition at line 209 of file TensorConvolution.h.


The documentation for this class was generated from the following file:


gtsam
Author(s):
autogenerated on Sun Dec 22 2024 04:21:55