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 ¶m) |
template<typename WeiLayout > | |
CK_TILE_HOST HostTensorDescriptor | make_weight_host_tensor_descriptor_g_k_c_xs_packed (const ck_tile::conv::ConvParam ¶m) |
template<typename OutLayout > | |
CK_TILE_HOST HostTensorDescriptor | make_output_host_tensor_descriptor_g_n_k_wos_packed (const ck_tile::conv::ConvParam ¶m) |
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[] | ||
) |