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

• develop
• include
• ck_tile
• ref

Classes | Namespaces | Functions

naive_grouped_conv_fwd_gpu.hpp File Reference

#include "ck_tile/core.hpp"
#include "ck_tile/ref/conv_common.hpp"
#include <array>
#include "ck_tile/host/device_memory.hpp"
#include "ck_tile/host/kernel_launch.hpp"
#include <hip/hip_runtime.h>

Go to the source code of this file.

Classes
struct	ck_tile::naive_grouped_conv_fwd_kernel< NDimSpatial, InDataType, WeiDataType, OutDataType >

Namespaces
	ck_tile

Functions
template<ck_tile::index_t NDimSpatial, typename InDataType , typename WeiDataType , typename OutDataType >
CK_TILE_HOST float	ck_tile::naive_grouped_conv_fwd (const InDataType *p_in_dev, const WeiDataType *p_wei_dev, OutDataType *p_out_dev, ck_tile::index_t G, ck_tile::index_t N, ck_tile::index_t K, ck_tile::index_t C, std::vector< ck_tile::long_index_t > in_spatial_lengths, std::vector< ck_tile::long_index_t > wei_spatial_lengths, std::vector< ck_tile::long_index_t > out_spatial_lengths, std::vector< ck_tile::long_index_t > conv_strides, std::vector< ck_tile::long_index_t > conv_dilations, std::vector< ck_tile::long_index_t > in_left_pads, ck_tile::stream_config stream_config={})