TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType > Struct Template Reference#
ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType > Struct Template Reference
#include <transform_conv_bwd_weight_to_gemm.hpp>
Public Member Functions | |
| CK_TILE_HOST constexpr | TransformConvBwdWeightToGemm () |
| template<typename TransformConvBwdWeightToGemmBase> | |
| CK_TILE_HOST | TransformConvBwdWeightToGemm (const TransformConvBwdWeightToGemmBase &transform_conv_fwd_to_gemm_base) |
| template<typename ConvDimsType, typename ConvSpatialDimsType, index_t NDim = NDimSpatial, typename std::enable_if< NDim==1, bool >::type = false> | |
| CK_TILE_HOST | TransformConvBwdWeightToGemm (const ConvDimsType &a_g_n_c_wis_lengths, const ConvDimsType &b_g_k_c_xs_lengths, const ConvDimsType &c_g_n_k_wos_lengths, const ConvSpatialDimsType &conv_filter_strides, const ConvSpatialDimsType &conv_filter_dilations, const ConvSpatialDimsType &input_left_pads, const ConvSpatialDimsType &input_right_pads) |
| template<typename ConvDimsType, typename ConvSpatialDimsType, index_t NDim = NDimSpatial, typename std::enable_if< NDim==2, bool >::type = false> | |
| CK_TILE_HOST | TransformConvBwdWeightToGemm (const ConvDimsType &a_g_n_c_wis_lengths, const ConvDimsType &b_g_k_c_xs_lengths, const ConvDimsType &c_g_n_k_wos_lengths, const ConvSpatialDimsType &conv_filter_strides, const ConvSpatialDimsType &conv_filter_dilations, const ConvSpatialDimsType &input_left_pads, const ConvSpatialDimsType &input_right_pads) |
| template<typename ConvDimsType, typename ConvSpatialDimsType, index_t NDim = NDimSpatial, typename std::enable_if< NDim==3, bool >::type = false> | |
| CK_TILE_HOST | TransformConvBwdWeightToGemm (const ConvDimsType &a_g_n_c_wis_lengths, const ConvDimsType &b_g_k_c_xs_lengths, const ConvDimsType &c_g_n_k_wos_lengths, const ConvSpatialDimsType &conv_filter_strides, const ConvSpatialDimsType &conv_filter_dilations, const ConvSpatialDimsType &input_left_pads, const ConvSpatialDimsType &input_right_pads) |
| template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==1, bool >::type = false> | |
| CK_TILE_HOST auto | make_out_grid_desc () const |
| template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==1, bool >::type = false> | |
| CK_TILE_HOST auto | make_in_grid_desc () const |
| template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==1, bool >::type = false> | |
| CK_TILE_HOST auto | make_wei_grid_desc () const |
| template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==2, bool >::type = false> | |
| CK_TILE_HOST auto | make_out_grid_desc () const |
| template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==2, bool >::type = false> | |
| CK_TILE_HOST auto | make_in_grid_desc () const |
| template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==2, bool >::type = false> | |
| CK_TILE_HOST auto | make_wei_grid_desc () const |
| template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==3, bool >::type = false> | |
| CK_TILE_HOST auto | make_out_grid_desc () const |
| template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==3, bool >::type = false> | |
| CK_TILE_HOST auto | make_in_grid_desc () const |
| template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==3, bool >::type = false> | |
| CK_TILE_HOST auto | make_wei_grid_desc () const |
| template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==1, bool >::type = false> | |
| CK_TILE_HOST auto | MakeABCGridDescriptor_A_K0_M_K1_B_K0_N_K1_C_M_N () const |
| template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==2, bool >::type = false> | |
| CK_TILE_HOST auto | MakeABCGridDescriptor_A_K0_M_K1_B_K0_N_K1_C_M_N () const |
| template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==3, bool >::type = false> | |
| CK_TILE_HOST auto | MakeABCGridDescriptor_A_K0_M_K1_B_K0_N_K1_C_M_N () const |
Public Attributes | |
| IndexType | G_ |
| IndexType | N_ |
| IndexType | Di_ |
| IndexType | Hi_ |
| IndexType | Wi_ |
| IndexType | Do_ |
| IndexType | Ho_ |
| IndexType | Wo_ |
| IndexType | Z_ |
| IndexType | Y_ |
| IndexType | X_ |
| IndexType | K_ |
| IndexType | C_ |
| IndexType | ConvStrideD_ |
| IndexType | ConvStrideH_ |
| IndexType | ConvStrideW_ |
| IndexType | ConvDilationD_ |
| IndexType | ConvDilationH_ |
| IndexType | ConvDilationW_ |
| IndexType | InLeftPadD_ |
| IndexType | InLeftPadH_ |
| IndexType | InLeftPadW_ |
| IndexType | InRightPadD_ |
| IndexType | InRightPadH_ |
| IndexType | InRightPadW_ |
Constructor & Destructor Documentation
◆ TransformConvBwdWeightToGemm() [1/5]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
|
inlineconstexpr |
◆ TransformConvBwdWeightToGemm() [2/5]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
template<typename TransformConvBwdWeightToGemmBase>
|
inline |
◆ TransformConvBwdWeightToGemm() [3/5]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
template<typename ConvDimsType, typename ConvSpatialDimsType, index_t NDim = NDimSpatial, typename std::enable_if< NDim==1, bool >::type = false>
|
inline |
◆ TransformConvBwdWeightToGemm() [4/5]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
template<typename ConvDimsType, typename ConvSpatialDimsType, index_t NDim = NDimSpatial, typename std::enable_if< NDim==2, bool >::type = false>
|
inline |
◆ TransformConvBwdWeightToGemm() [5/5]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
template<typename ConvDimsType, typename ConvSpatialDimsType, index_t NDim = NDimSpatial, typename std::enable_if< NDim==3, bool >::type = false>
|
inline |
Member Function Documentation
◆ make_in_grid_desc() [1/3]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==3, bool >::type = false>
|
inline |
◆ make_in_grid_desc() [2/3]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==2, bool >::type = false>
|
inline |
◆ make_in_grid_desc() [3/3]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==1, bool >::type = false>
|
inline |
◆ make_out_grid_desc() [1/3]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==3, bool >::type = false>
|
inline |
◆ make_out_grid_desc() [2/3]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==2, bool >::type = false>
|
inline |
◆ make_out_grid_desc() [3/3]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==1, bool >::type = false>
|
inline |
◆ make_wei_grid_desc() [1/3]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==3, bool >::type = false>
|
inline |
◆ make_wei_grid_desc() [2/3]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==2, bool >::type = false>
|
inline |
◆ make_wei_grid_desc() [3/3]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==1, bool >::type = false>
|
inline |
◆ MakeABCGridDescriptor_A_K0_M_K1_B_K0_N_K1_C_M_N() [1/3]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==3, bool >::type = false>
|
inline |
◆ MakeABCGridDescriptor_A_K0_M_K1_B_K0_N_K1_C_M_N() [2/3]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==2, bool >::type = false>
|
inline |
◆ MakeABCGridDescriptor_A_K0_M_K1_B_K0_N_K1_C_M_N() [3/3]
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
template<index_t NDim = NDimSpatial, typename std::enable_if< NDim==1, bool >::type = false>
|
inline |
Member Data Documentation
◆ C_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::C_ |
◆ ConvDilationD_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::ConvDilationD_ |
◆ ConvDilationH_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::ConvDilationH_ |
◆ ConvDilationW_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::ConvDilationW_ |
◆ ConvStrideD_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::ConvStrideD_ |
◆ ConvStrideH_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::ConvStrideH_ |
◆ ConvStrideW_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::ConvStrideW_ |
◆ Di_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::Di_ |
◆ Do_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::Do_ |
◆ G_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::G_ |
◆ Hi_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::Hi_ |
◆ Ho_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::Ho_ |
◆ InLeftPadD_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::InLeftPadD_ |
◆ InLeftPadH_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::InLeftPadH_ |
◆ InLeftPadW_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::InLeftPadW_ |
◆ InRightPadD_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::InRightPadD_ |
◆ InRightPadH_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::InRightPadH_ |
◆ InRightPadW_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::InRightPadW_ |
◆ K_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::K_ |
◆ N_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::N_ |
◆ Wi_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::Wi_ |
◆ Wo_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::Wo_ |
◆ X_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::X_ |
◆ Y_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::Y_ |
◆ Z_
template<index_t NDimSpatial, ConvolutionSpecialization ConvolutionSpecialization, 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>
| IndexType ck_tile::TransformConvBwdWeightToGemm< NDimSpatial, ConvolutionSpecialization, VectorSizeA, VectorSizeB, VectorSizeC, NumGroupsToMerge, SplitN, ADataType, CDataType, IndexType >::Z_ |
The documentation for this struct was generated from the following file: