/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ref/conv_common.hpp File Reference

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ref/conv_common.hpp File Reference#

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ref/conv_common.hpp File Reference
Namespaces | Functions
conv_common.hpp File Reference
#include "ck_tile/core.hpp"
#include <array>
#include <vector>

Go to the source code of this file.

Namespaces

 ck_tile
 
 ck_tile::detail
 

Functions

template<ck_tile::index_t NDimSpatial>
std::array< ck_tile::long_index_t, NDimSpatial > ck_tile::to_array (const std::vector< ck_tile::long_index_t > &vec)
 
template<ck_tile::index_t NDimSpatial>
std::array< ck_tile::long_index_t, NDimSpatial > ck_tile::to_array_with_default (const std::vector< ck_tile::long_index_t > &vec, ck_tile::long_index_t default_val=1)
 
template<index_t NDimSpatial>
__device__ long_index_t ck_tile::detail::calculate_input_index (index_t n, index_t g, index_t c, const std::array< index_t, NDimSpatial > &spatial_idx, const std::array< long_index_t, NDimSpatial+3 > &strides)
 
template<index_t NDimSpatial>
__device__ long_index_t ck_tile::detail::calculate_weight_index (index_t g, index_t k, index_t c, const std::array< index_t, NDimSpatial > &spatial_idx, const std::array< long_index_t, NDimSpatial+3 > &strides)
 
template<index_t NDimSpatial>
__device__ long_index_t ck_tile::detail::calculate_output_index (index_t n, index_t g, index_t k, const std::array< index_t, NDimSpatial > &spatial_idx, const std::array< long_index_t, NDimSpatial+3 > &strides)