include/ck_tile/host/reference/reference_im2col.hpp File Reference

include/ck_tile/host/reference/reference_im2col.hpp File Reference#

Composable Kernel: include/ck_tile/host/reference/reference_im2col.hpp File Reference
reference_im2col.hpp File Reference
#include "ck_tile/core.hpp"
#include "ck_tile/host/host_tensor.hpp"
#include <thread>

Go to the source code of this file.

Namespaces

 ck_tile
 

Functions

template<typename InDataType , typename OutDataType , index_t NDimSpatial>
CK_TILE_HOST void ck_tile::reference_im2col (const HostTensor< InDataType > &in_host, HostTensor< OutDataType > &out_host, const ck_tile::conv::ConvParam &conv_params)