SplitImageInfo Struct Reference

• ck_tile
• TransformConvFwdToGemm
• SplitImageInfo

Public Attributes | List of all members

ck_tile::TransformConvFwdToGemm< NDimSpatial, ConvSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::SplitImageInfo Struct Reference

#include <transform_conv_fwd_to_gemm.hpp>

Public Attributes
bool	should_split
index_t	num_d_pieces
index_t	num_h_pieces
index_t	num_w_pieces

Member Data Documentation

◆ num_d_pieces

template<index_t NDimSpatial, ConvolutionSpecialization ConvSpecialization, index_t VectorSizeA, index_t VectorSizeB, index_t VectorSizeC, index_t NumGroupsToMerge = 1, bool SplitN = false, typename ADataType = float, typename CDataType = float, typename IndexType = index_t>

index_t ck_tile::TransformConvFwdToGemm< NDimSpatial, ConvSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::SplitImageInfo::num_d_pieces

◆ num_h_pieces

template<index_t NDimSpatial, ConvolutionSpecialization ConvSpecialization, index_t VectorSizeA, index_t VectorSizeB, index_t VectorSizeC, index_t NumGroupsToMerge = 1, bool SplitN = false, typename ADataType = float, typename CDataType = float, typename IndexType = index_t>

index_t ck_tile::TransformConvFwdToGemm< NDimSpatial, ConvSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::SplitImageInfo::num_h_pieces

◆ num_w_pieces

template<index_t NDimSpatial, ConvolutionSpecialization ConvSpecialization, index_t VectorSizeA, index_t VectorSizeB, index_t VectorSizeC, index_t NumGroupsToMerge = 1, bool SplitN = false, typename ADataType = float, typename CDataType = float, typename IndexType = index_t>

index_t ck_tile::TransformConvFwdToGemm< NDimSpatial, ConvSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::SplitImageInfo::num_w_pieces

◆ should_split

template<index_t NDimSpatial, ConvolutionSpecialization ConvSpecialization, index_t VectorSizeA, index_t VectorSizeB, index_t VectorSizeC, index_t NumGroupsToMerge = 1, bool SplitN = false, typename ADataType = float, typename CDataType = float, typename IndexType = index_t>

bool ck_tile::TransformConvFwdToGemm< NDimSpatial, ConvSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::SplitImageInfo::should_split

---

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

• tile/ops/grouped_convolution/utils/transform_conv_fwd_to_gemm.hpp