|
auto | layout_type (std::string type) |
|
ck::tensor_operation::device::GemmSpecialization | gemm_type (std::string type) |
|
ck::tensor_operation::device::ConvolutionForwardSpecialization | conv_type (std::string type) |
|
template<typename CDesc_MRaw_NRaw > |
auto | pad (ck::index_t mpb, ck::index_t npb, ck::index_t kpb, ck::tensor_operation::device::GemmSpecialization gemm, CDesc_MRaw_NRaw conv) |
|
auto | transform_conv (ck::index_t num_dim, ck::tensor_operation::device::ConvolutionForwardSpecialization spec, ck::Array< ck::index_t, 5 > out_lengths, ck::Array< ck::index_t, 5 > out_strides) |
|
auto | transform_conv_3d (ck::index_t num_dim, ck::tensor_operation::device::ConvolutionForwardSpecialization spec, ck::Array< ck::index_t, 6 > out_lengths, ck::Array< ck::index_t, 6 > out_strides) |
|
auto | transform_conv_1d (ck::index_t num_dim, ck::tensor_operation::device::ConvolutionForwardSpecialization spec, ck::Array< ck::index_t, 4 > out_lengths, ck::Array< ck::index_t, 4 > out_strides) |
|
template<typename CGridDesc_M_N > |
auto | block_2_etile (ck::index_t m_per_block, ck::index_t n_per_block, CGridDesc_M_N matrix_padder) |
|
auto | get_launch_params_1d (ck::host::Solution solution, ck::Array< ck::index_t, 4 > out_lengths, ck::Array< ck::index_t, 4 > out_strides) |
|
auto | get_launch_params (ck::host::Solution solution, ck::Array< ck::index_t, 5 > out_lengths, ck::Array< ck::index_t, 5 > out_strides) |
|
auto | get_launch_params_3d (ck::host::Solution solution, ck::Array< ck::index_t, 6 > out_lengths, ck::Array< ck::index_t, 6 > out_strides) |
|