/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 File Reference

• develop
• include
• ck_tile
• ops
• grouped_convolution
• utils

Classes | Namespaces | Typedefs | Enumerations | Functions

grouped_convolution_utils.hpp File Reference

#include "ck_tile/core.hpp"
#include "ck_tile/host/convolution_parameter.hpp"
#include "ck_tile/ops/elementwise/unary_element_wise_operation.hpp"

Go to the source code of this file.

Classes
struct	ck_tile::GroupedConvHostArgs< InPtr, WeiPtr, OutPtr, CDElementwise >
	The Grouped Conv kernel host arguments. More...
struct	ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_, ExplicitGemm_ >
struct	ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_, ExplicitGemm_ >::FixedGemmParams
struct	ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_, ExplicitGemm_ >::GemmLayouts< Direction >
struct	ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_, ExplicitGemm_ >::GemmLayouts< GroupedConvDirection::FORWARD >
struct	ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_, ExplicitGemm_ >::GemmLayouts< GroupedConvDirection::BACKWARD_DATA >
struct	ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_, ExplicitGemm_ >::GemmLayouts< GroupedConvDirection::BACKWARD_WEIGHT >
struct	ck_tile::SplitImagePieceInfo
	Helper struct for split-image piece information. More...

Namespaces
	ck_tile

Typedefs
using	ck_tile::PassThrough = ck_tile::element_wise::PassThrough
template<typename CDElementwise = PassThrough>
using	ck_tile::GroupedConvFwdHostArgs = GroupedConvHostArgs< const void *, const void *, void *, CDElementwise >
using	ck_tile::GroupedConvBwdWeightHostArgs = GroupedConvHostArgs< const void *, void *, const void *, PassThrough >
using	ck_tile::GroupedConvBwdDataHostArgs = GroupedConvHostArgs< void *, const void *, const void *, PassThrough >

Enumerations
enum class	ck_tile::GroupedConvDirection {   ck_tile::FORWARD ,   ck_tile::BACKWARD_DATA ,   ck_tile::BACKWARD_WEIGHT }

Functions
template<typename TilePartitioner >
CK_TILE_HOST SplitImagePieceInfo	ck_tile::calculate_spatial_piece (ck_tile::index_t piece_idx, ck_tile::index_t num_d_pieces, ck_tile::index_t num_h_pieces, ck_tile::index_t num_w_pieces, ck_tile::index_t base_piece_d, ck_tile::index_t base_piece_h, ck_tile::index_t base_piece_w, ck_tile::index_t total_d, ck_tile::index_t total_h, ck_tile::index_t total_w, ck_tile::index_t N, ck_tile::index_t K, ck_tile::index_t total_blocks)
	Calculate piece information for split-image convolution. More...