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 mapCudaInputKernelToTensorInputOffset (Index i) const
 
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index mapCudaInputKernelToTensorInputOffset (Index i, Index j) const
 
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index mapCudaInputKernelToTensorInputOffset (Index i, Index j, Index k) const
 
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index mapCudaInputPlaneToTensorInputOffset (Index p) const
 
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index mapCudaOutputKernelToTensorOutputOffset (Index i) const
 
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index mapCudaOutputKernelToTensorOutputOffset (Index i, Index j) const
 
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index mapCudaOutputKernelToTensorOutputOffset (Index i, Index j, Index k) const
 
EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC Index mapCudaOutputPlaneToTensorOutputOffset (Index p) const
 

Private Attributes

array< Index, NumDimsm_cudaInputStrides
 
array< Index, NumDimsm_cudaOutputStrides
 
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

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

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

Definition at line 164 of file TensorConvolution.h.

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

Definition at line 178 of file TensorConvolution.h.

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

Definition at line 192 of file TensorConvolution.h.

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

Definition at line 116 of file TensorConvolution.h.

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

Definition at line 171 of file TensorConvolution.h.

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

Definition at line 185 of file TensorConvolution.h.

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

Definition at line 200 of file TensorConvolution.h.

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

Definition at line 140 of file TensorConvolution.h.

Member Data Documentation

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

Definition at line 212 of file TensorConvolution.h.

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

Definition at line 213 of file TensorConvolution.h.

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.

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.

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:


hebiros
Author(s): Xavier Artache , Matthew Tesch
autogenerated on Thu Sep 3 2020 04:10:37