#include <grouped_convolution_utils.hpp>
|
using | InLayout = InLayout_ |
|
using | WeiLayout = WeiLayout_ |
|
using | DsLayout = DsLayout_ |
|
using | OutLayout = OutLayout_ |
|
using | GroupedConvImplicitGemmTraitsFwd = TileGemmTraits< true, true, true, ck_tile::tensor_layout::gemm::RowMajor, ck_tile::tensor_layout::gemm::ColumnMajor, ck_tile::tensor_layout::gemm::RowMajor > |
|
using | GroupedConvImplicitGemmTraitsBwdData = TileGemmTraits< true, true, true, ck_tile::tensor_layout::gemm::RowMajor, ck_tile::tensor_layout::gemm::ColumnMajor, ck_tile::tensor_layout::gemm::RowMajor > |
|
using | GroupedConvImplicitGemmTraitsBwdWeight = TileGemmTraits< true, true, true, ck_tile::tensor_layout::gemm::RowMajor, ck_tile::tensor_layout::gemm::ColumnMajor, ck_tile::tensor_layout::gemm::RowMajor > |
|
using | ImplicitGemmDsLayout = decltype(generate_implicit_gemm_layout()) |
|
◆ DsLayout
template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_ , typename WeiLayout_ , typename DsLayout_ , typename OutLayout_ , index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1>
using ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_ >::DsLayout = DsLayout_ |
◆ GroupedConvImplicitGemmTraitsBwdData
template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_ , typename WeiLayout_ , typename DsLayout_ , typename OutLayout_ , index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1>
using ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_ >::GroupedConvImplicitGemmTraitsBwdData = TileGemmTraits<true, true, true, ck_tile::tensor_layout::gemm::RowMajor, ck_tile::tensor_layout::gemm::ColumnMajor, ck_tile::tensor_layout::gemm::RowMajor> |
◆ GroupedConvImplicitGemmTraitsBwdWeight
template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_ , typename WeiLayout_ , typename DsLayout_ , typename OutLayout_ , index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1>
using ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_ >::GroupedConvImplicitGemmTraitsBwdWeight = TileGemmTraits<true, true, true, ck_tile::tensor_layout::gemm::RowMajor, ck_tile::tensor_layout::gemm::ColumnMajor, ck_tile::tensor_layout::gemm::RowMajor> |
◆ GroupedConvImplicitGemmTraitsFwd
template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_ , typename WeiLayout_ , typename DsLayout_ , typename OutLayout_ , index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1>
using ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_ >::GroupedConvImplicitGemmTraitsFwd = TileGemmTraits<true, true, true, ck_tile::tensor_layout::gemm::RowMajor, ck_tile::tensor_layout::gemm::ColumnMajor, ck_tile::tensor_layout::gemm::RowMajor> |
◆ ImplicitGemmDsLayout
template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_ , typename WeiLayout_ , typename DsLayout_ , typename OutLayout_ , index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1>
◆ InLayout
template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_ , typename WeiLayout_ , typename DsLayout_ , typename OutLayout_ , index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1>
using ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_ >::InLayout = InLayout_ |
◆ OutLayout
template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_ , typename WeiLayout_ , typename DsLayout_ , typename OutLayout_ , index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1>
using ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_ >::OutLayout = OutLayout_ |
◆ WeiLayout
template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_ , typename WeiLayout_ , typename DsLayout_ , typename OutLayout_ , index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1>
using ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_ >::WeiLayout = WeiLayout_ |
◆ ConvSpecialization
template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_ , typename WeiLayout_ , typename DsLayout_ , typename OutLayout_ , index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1>
◆ NDimSpatial
template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_ , typename WeiLayout_ , typename DsLayout_ , typename OutLayout_ , index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1>
constexpr index_t ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_ >::NDimSpatial = NDimSpatial_ |
|
staticconstexpr |
◆ NumDTensor
template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_ , typename WeiLayout_ , typename DsLayout_ , typename OutLayout_ , index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1>
◆ NumGroupsToMerge
template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_ , typename WeiLayout_ , typename DsLayout_ , typename OutLayout_ , index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1>
constexpr index_t ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_ >::NumGroupsToMerge = 1 |
|
staticconstexpr |
◆ VectorSizeA
template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_ , typename WeiLayout_ , typename DsLayout_ , typename OutLayout_ , index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1>
◆ VectorSizeB
template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_ , typename WeiLayout_ , typename DsLayout_ , typename OutLayout_ , index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1>
◆ VectorSizeC
template<index_t NDimSpatial_, ConvolutionSpecialization ConvSpecialization_, typename InLayout_ , typename WeiLayout_ , typename DsLayout_ , typename OutLayout_ , index_t VectorSizeA_ = 1, index_t VectorSizeB_ = 1, index_t VectorSizeC_ = 1>
The documentation for this struct was generated from the following file:
- /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/grouped_convolution/utils/grouped_convolution_utils.hpp