#include <image_to_column_kernel.hpp>
|
static constexpr CK_TILE_HOST Kargs | MakeKargs (const void *p_in, void *p_out, const long_index_t G, const long_index_t N, const long_index_t C, const array< long_index_t, NDimSpatial > input_spatial_lengths, const array< long_index_t, NDimSpatial > filter_spatial_lengths, const array< long_index_t, NDimSpatial > output_spatial_lengths, const array< long_index_t, NDimSpatial+3 > image_g_n_c_wis_strides, const array< long_index_t, 3 > gemm_g_m_k_strides, const array< long_index_t, NDimSpatial > conv_filter_strides, const array< long_index_t, NDimSpatial > conv_filter_dilations, const array< long_index_t, NDimSpatial > input_left_pads, const array< long_index_t, NDimSpatial > input_right_pads) |
|
static constexpr CK_TILE_HOST auto | GridSize (index_t GemmM, index_t GemmK, index_t Batch) |
|
static constexpr CK_TILE_HOST auto | BlockSize () |
|
static constexpr CK_TILE_DEVICE auto | MakeBlockTileDistribution () |
|
◆ InDataType
template<typename Problem_ >
◆ OutDataType
template<typename Problem_ >
◆ Problem
template<typename Problem_ >
◆ BlockSize()
template<typename Problem_ >
◆ CalculateMKDims()
template<typename Problem_ >
◆ ConvTensorRearrange()
template<typename Problem_ >
◆ GridSize()
template<typename Problem_ >
◆ MakeBlockTileDistribution()
template<typename Problem_ >
◆ MakeImageMKDesc()
template<typename Problem_ >
◆ MakeKargs()
template<typename Problem_ >
static constexpr CK_TILE_HOST Kargs ck_tile::ImageToColumn< Problem_ >::MakeKargs |
( |
const void * |
p_in, |
|
|
void * |
p_out, |
|
|
const long_index_t |
G, |
|
|
const long_index_t |
N, |
|
|
const long_index_t |
C, |
|
|
const array< long_index_t, NDimSpatial > |
input_spatial_lengths, |
|
|
const array< long_index_t, NDimSpatial > |
filter_spatial_lengths, |
|
|
const array< long_index_t, NDimSpatial > |
output_spatial_lengths, |
|
|
const array< long_index_t, NDimSpatial+3 > |
image_g_n_c_wis_strides, |
|
|
const array< long_index_t, 3 > |
gemm_g_m_k_strides, |
|
|
const array< long_index_t, NDimSpatial > |
conv_filter_strides, |
|
|
const array< long_index_t, NDimSpatial > |
conv_filter_dilations, |
|
|
const array< long_index_t, NDimSpatial > |
input_left_pads, |
|
|
const array< long_index_t, NDimSpatial > |
input_right_pads |
|
) |
| |
|
inlinestaticconstexpr |
◆ operator()()
template<typename Problem_ >
◆ AligmentIn
template<typename Problem_ >
◆ AligmentOut
template<typename Problem_ >
◆ I0
template<typename Problem_ >
◆ I1
template<typename Problem_ >
◆ I2
template<typename Problem_ >
◆ I3
template<typename Problem_ >
◆ I4
template<typename Problem_ >
◆ kKPerBlock
template<typename Problem_ >
◆ kMPerBlock
template<typename Problem_ >
◆ NDimSpatial
template<typename Problem_ >
The documentation for this struct was generated from the following file:
- /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck_tile/ops/image_to_column/kernel/image_to_column_kernel.hpp