transform_forward_convolution3d_into_gemm_v4r4r4_ndhwc_kzyxc_ndhwk.hpp File Reference#
transform_forward_convolution3d_into_gemm_v4r4r4_ndhwc_kzyxc_ndhwk.hpp File Reference
#include "common_header.hpp"#include "tensor_descriptor.hpp"#include "tensor_descriptor_helper.hpp"Go to the source code of this file.
Namespaces | |
| namespace | ck |
Functions | |
| template<typename... In, typename... Wei, typename... Out, typename ConvStrides, typename ConvDilations, typename InLeftPads, typename InRightPads, index_t GemmK1Value> | |
| __host__ __device__ constexpr auto | ck::transform_forward_convolution3d_into_gemm_v4r4r4_ndhwc_kzyxc_ndhwk_pad (const TensorDescriptor< In... > &in_grid_desc_n_di_hi_wi_c, const TensorDescriptor< Wei... > &wei_k_z_y_x_c_grid_desc, const TensorDescriptor< Out... > &out_n_do_ho_wo_k_grid_desc, const ConvStrides &conv_strides, const ConvDilations &conv_dilations, const InLeftPads &in_left_pads, const InRightPads &in_right_pads, Number< GemmK1Value >) |