reference_im2col.hpp File Reference

reference_im2col.hpp File Reference#

Composable Kernel: 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

namespace  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)