grouped_convolution_utils.hpp File Reference

• include
• ck_tile
• ops
• grouped_convolution
• utils

Classes | Namespaces | Typedefs | 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_ >
struct	ck_tile::GroupedConvTraits< NDimSpatial_, ConvSpecialization_, InLayout_, WeiLayout_, DsLayout_, OutLayout_, VectorSizeA_, VectorSizeB_, VectorSizeC_, NumGroupsToMerge_, EnableSplitImage_ >::FixedGemmParams
struct	ck_tile::SplitImagePieceInfo
	Helper struct for split-image piece information. More...

Namespaces
namespace	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
using	ck_tile::GroupedConvBwdDataHostArgs

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.