SplitImageInfo Struct Reference

SplitImageInfo Struct Reference#

Composable Kernel: ck_tile::GroupedConvFwdKernelArgs< GroupedConvTraitsType_, CDElementwise_ >::SplitImageInfo Struct Reference
Classes | Public Attributes | Static Public Attributes | List of all members
ck_tile::GroupedConvFwdKernelArgs< GroupedConvTraitsType_, CDElementwise_ >::SplitImageInfo Struct Reference

#include <grouped_convolution_forward_kernel.hpp>

Classes

struct  PieceInfo

Public Attributes

index_t total_d = 1
index_t total_h = 1
index_t total_w = 1
index_t total_spatial = 1
index_t num_d_pieces = 1
index_t num_h_pieces = 1
index_t num_w_pieces = 1
std::array< PieceInfo, MaxPieces > pieces

Static Public Attributes

static constexpr index_t MaxPieces = 64

Member Data Documentation

◆ MaxPieces

template<typename GroupedConvTraitsType_, typename CDElementwise_>
index_t ck_tile::GroupedConvFwdKernelArgs< GroupedConvTraitsType_, CDElementwise_ >::SplitImageInfo::MaxPieces = 64
staticconstexpr

◆ num_d_pieces

template<typename GroupedConvTraitsType_, typename CDElementwise_>
index_t ck_tile::GroupedConvFwdKernelArgs< GroupedConvTraitsType_, CDElementwise_ >::SplitImageInfo::num_d_pieces = 1

◆ num_h_pieces

template<typename GroupedConvTraitsType_, typename CDElementwise_>
index_t ck_tile::GroupedConvFwdKernelArgs< GroupedConvTraitsType_, CDElementwise_ >::SplitImageInfo::num_h_pieces = 1

◆ num_w_pieces

template<typename GroupedConvTraitsType_, typename CDElementwise_>
index_t ck_tile::GroupedConvFwdKernelArgs< GroupedConvTraitsType_, CDElementwise_ >::SplitImageInfo::num_w_pieces = 1

◆ pieces

template<typename GroupedConvTraitsType_, typename CDElementwise_>
std::array<PieceInfo, MaxPieces> ck_tile::GroupedConvFwdKernelArgs< GroupedConvTraitsType_, CDElementwise_ >::SplitImageInfo::pieces

◆ total_d

template<typename GroupedConvTraitsType_, typename CDElementwise_>
index_t ck_tile::GroupedConvFwdKernelArgs< GroupedConvTraitsType_, CDElementwise_ >::SplitImageInfo::total_d = 1

◆ total_h

template<typename GroupedConvTraitsType_, typename CDElementwise_>
index_t ck_tile::GroupedConvFwdKernelArgs< GroupedConvTraitsType_, CDElementwise_ >::SplitImageInfo::total_h = 1

◆ total_spatial

template<typename GroupedConvTraitsType_, typename CDElementwise_>
index_t ck_tile::GroupedConvFwdKernelArgs< GroupedConvTraitsType_, CDElementwise_ >::SplitImageInfo::total_spatial = 1

◆ total_w

template<typename GroupedConvTraitsType_, typename CDElementwise_>
index_t ck_tile::GroupedConvFwdKernelArgs< GroupedConvTraitsType_, CDElementwise_ >::SplitImageInfo::total_w = 1
The documentation for this struct was generated from the following file: