conv Namespace Reference

conv Namespace Reference#

Composable Kernel: ck_tile::conv Namespace Reference
ck_tile::conv Namespace Reference

Namespaces

 detail
 

Classes

struct  ConvParam
 

Functions

template<typename InLayout >
CK_TILE_HOST HostTensorDescriptor make_input_host_tensor_descriptor_g_n_c_wis_packed (const ck_tile::conv::ConvParam &param)
 
template<typename WeiLayout >
CK_TILE_HOST HostTensorDescriptor make_weight_host_tensor_descriptor_g_k_c_xs_packed (const ck_tile::conv::ConvParam &param)
 
template<typename OutLayout >
CK_TILE_HOST HostTensorDescriptor make_output_host_tensor_descriptor_g_n_k_wos_packed (const ck_tile::conv::ConvParam &param)
 
CK_TILE_HOST std::string get_conv_param_parser_helper_msg ()
 
CK_TILE_HOST ck_tile::conv::ConvParam parse_conv_param (int num_dim_spatial, int arg_idx, char *const argv[])
 

Function Documentation

◆ get_conv_param_parser_helper_msg()

CK_TILE_HOST std::string ck_tile::conv::get_conv_param_parser_helper_msg ( )

◆ make_input_host_tensor_descriptor_g_n_c_wis_packed()

template<typename InLayout >
CK_TILE_HOST HostTensorDescriptor ck_tile::conv::make_input_host_tensor_descriptor_g_n_c_wis_packed ( const ck_tile::conv::ConvParam param)

◆ make_output_host_tensor_descriptor_g_n_k_wos_packed()

template<typename OutLayout >
CK_TILE_HOST HostTensorDescriptor ck_tile::conv::make_output_host_tensor_descriptor_g_n_k_wos_packed ( const ck_tile::conv::ConvParam param)

◆ make_weight_host_tensor_descriptor_g_k_c_xs_packed()

template<typename WeiLayout >
CK_TILE_HOST HostTensorDescriptor ck_tile::conv::make_weight_host_tensor_descriptor_g_k_c_xs_packed ( const ck_tile::conv::ConvParam param)

◆ parse_conv_param()

CK_TILE_HOST ck_tile::conv::ConvParam ck_tile::conv::parse_conv_param ( int  num_dim_spatial,
int  arg_idx,
char *const  argv[] 
)