convolution_host_tensor_descriptor_helper.hpp File Reference

convolution_host_tensor_descriptor_helper.hpp File Reference#

Composable Kernel: convolution_host_tensor_descriptor_helper.hpp File Reference
Namespaces | Functions
convolution_host_tensor_descriptor_helper.hpp File Reference
#include "ck_tile/ops/common/tensor_layout.hpp"
#include "ck_tile/host/convolution_parameter.hpp"
#include "ck_tile/host/host_tensor.hpp"

Go to the source code of this file.

Namespaces

namespace  ck_tile
namespace  ck_tile::conv
namespace  ck_tile::conv::detail

Functions

template<typename OldLayout>
CK_TILE_HOST std::vector< std::size_t > ck_tile::conv::detail::get_layout_transpose_gnchw_to_old ()
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)
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)
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)