conv Namespace Reference

• ck
• utils
• conv

Namespaces | Classes | Functions

ck::utils::conv Namespace Reference

Namespaces
	detail

Classes
struct	ConvParam

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

Function Documentation

◆ get_conv_param_parser_helper_msg()

std::string ck::utils::conv::get_conv_param_parser_helper_msg

(

)

◆ make_input_host_tensor_descriptor_g_n_c_wis_packed()

template<typename InLayout >

HostTensorDescriptor ck::utils::conv::make_input_host_tensor_descriptor_g_n_c_wis_packed

(

const ck::utils::conv::ConvParam &

param

)

◆ make_output_host_tensor_descriptor_g_n_k_wos_packed()

template<typename OutLayout >

HostTensorDescriptor ck::utils::conv::make_output_host_tensor_descriptor_g_n_k_wos_packed

(

const ck::utils::conv::ConvParam &

param

)

◆ make_weight_host_tensor_descriptor_g_k_c_xs_packed()

template<typename WeiLayout >

HostTensorDescriptor ck::utils::conv::make_weight_host_tensor_descriptor_g_k_c_xs_packed

(

const ck::utils::conv::ConvParam &

param

)

◆ parse_conv_param()

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