Member List#
ck::tensor_operation::TransformConvBwdWeightToGemm< NDimSpatial, MPerBlock, NPerBlock, GemmK1Number, K0PerBlock, ConvBackwardWeightSpecialization > Member List
This is the complete list of members for ck::tensor_operation::TransformConvBwdWeightToGemm< NDimSpatial, MPerBlock, NPerBlock, GemmK1Number, K0PerBlock, ConvBackwardWeightSpecialization >, including all inherited members.
| I0 | ck::tensor_operation::TransformConvBwdWeightToGemm< NDimSpatial, MPerBlock, NPerBlock, GemmK1Number, K0PerBlock, ConvBackwardWeightSpecialization > | static |
| I1 | ck::tensor_operation::TransformConvBwdWeightToGemm< NDimSpatial, MPerBlock, NPerBlock, GemmK1Number, K0PerBlock, ConvBackwardWeightSpecialization > | static |
| make_in_grid_desc(const index_t N, const index_t Hi, const index_t Wi, const index_t C, const std::array< index_t, NDimSpatial+3 > &input_strides) | ck::tensor_operation::TransformConvBwdWeightToGemm< NDimSpatial, MPerBlock, NPerBlock, GemmK1Number, K0PerBlock, ConvBackwardWeightSpecialization > | inlinestatic |
| make_in_grid_desc(const index_t N, const index_t Di, const index_t Hi, const index_t Wi, const index_t C, const std::array< index_t, NDimSpatial+3 > &input_strides) | ck::tensor_operation::TransformConvBwdWeightToGemm< NDimSpatial, MPerBlock, NPerBlock, GemmK1Number, K0PerBlock, ConvBackwardWeightSpecialization > | inlinestatic |
| make_out_grid_desc(const index_t N, const index_t Ho, const index_t Wo, const index_t K, const std::array< index_t, NDimSpatial+3 > &output_strides) | ck::tensor_operation::TransformConvBwdWeightToGemm< NDimSpatial, MPerBlock, NPerBlock, GemmK1Number, K0PerBlock, ConvBackwardWeightSpecialization > | inlinestatic |
| make_out_grid_desc(const index_t N, const index_t Do, const index_t Ho, const index_t Wo, const index_t K, const std::array< index_t, NDimSpatial+3 > &output_strides) | ck::tensor_operation::TransformConvBwdWeightToGemm< NDimSpatial, MPerBlock, NPerBlock, GemmK1Number, K0PerBlock, ConvBackwardWeightSpecialization > | inlinestatic |
| make_wei_grid_desc(const index_t K, const index_t Y, const index_t X, const index_t C, const std::array< index_t, NDimSpatial+3 > &weights_strides) | ck::tensor_operation::TransformConvBwdWeightToGemm< NDimSpatial, MPerBlock, NPerBlock, GemmK1Number, K0PerBlock, ConvBackwardWeightSpecialization > | inlinestatic |
| make_wei_grid_desc(const index_t K, const index_t Z, const index_t Y, const index_t X, const index_t C, const std::array< index_t, NDimSpatial+3 > &weights_strides) | ck::tensor_operation::TransformConvBwdWeightToGemm< NDimSpatial, MPerBlock, NPerBlock, GemmK1Number, K0PerBlock, ConvBackwardWeightSpecialization > | inlinestatic |
| MakeABCGridDescriptor_A_K0_M_K1_B_K0_N_K1_C_M_N(const index_t N, const index_t K, const index_t C, const std::array< index_t, NDimSpatial > &input_spatial_lengths, const std::array< index_t, NDimSpatial > &filter_spatial_lengths, const std::array< index_t, NDimSpatial > &output_spatial_lengths, const std::array< index_t, NDimSpatial+3 > &, const std::array< index_t, NDimSpatial+3 > &, const std::array< index_t, NDimSpatial+3 > &, const std::array< index_t, NDimSpatial > &conv_filter_strides, const std::array< index_t, NDimSpatial > &conv_filter_dilations, const std::array< index_t, NDimSpatial > &input_left_pads, const std::array< index_t, NDimSpatial > &input_right_pads, const index_t batch_k) | ck::tensor_operation::TransformConvBwdWeightToGemm< NDimSpatial, MPerBlock, NPerBlock, GemmK1Number, K0PerBlock, ConvBackwardWeightSpecialization > | inlinestatic |
| MakeABCGridDescriptor_A_K0_M_K1_B_K0_N_K1_C_M_N(const index_t N, const index_t K, const index_t C, const std::array< index_t, NDimSpatial > &input_spatial_lengths, const std::array< index_t, NDimSpatial > &filter_spatial_lengths, const std::array< index_t, NDimSpatial > &output_spatial_lengths, const std::array< index_t, NDimSpatial+3 > &input_strides, const std::array< index_t, NDimSpatial+3 > &weights_strides, const std::array< index_t, NDimSpatial+3 > &output_strides, const std::array< index_t, NDimSpatial > &conv_filter_strides, const std::array< index_t, NDimSpatial > &conv_filter_dilations, const std::array< index_t, NDimSpatial > &input_left_pads, const std::array< index_t, NDimSpatial > &input_right_pads, const index_t batch_k) | ck::tensor_operation::TransformConvBwdWeightToGemm< NDimSpatial, MPerBlock, NPerBlock, GemmK1Number, K0PerBlock, ConvBackwardWeightSpecialization > | inlinestatic |
| MakeABCGridDescriptor_A_K0_M_K1_B_K0_N_K1_C_M_N(const index_t N, const index_t K, const index_t C, const std::array< index_t, NDimSpatial > &input_spatial_lengths, const std::array< index_t, NDimSpatial > &filter_spatial_lengths, const std::array< index_t, NDimSpatial > &output_spatial_lengths, const std::array< index_t, NDimSpatial+3 > &input_strides, const std::array< index_t, NDimSpatial+3 > &weights_strides, const std::array< index_t, NDimSpatial+3 > &output_strides, const std::array< index_t, NDimSpatial > &conv_filter_strides, const std::array< index_t, NDimSpatial > &conv_filter_dilations, const std::array< index_t, NDimSpatial > &input_left_pads, const std::array< index_t, NDimSpatial > &input_right_pads, const index_t batch_k) | ck::tensor_operation::TransformConvBwdWeightToGemm< NDimSpatial, MPerBlock, NPerBlock, GemmK1Number, K0PerBlock, ConvBackwardWeightSpecialization > | inlinestatic |