matrix_padder.hpp File Reference

• include
• ck
• tensor_operation
• gpu
• device

Classes | Namespaces | Functions

matrix_padder.hpp File Reference

#include "ck/utility/common_header.hpp"
#include "ck/tensor_description/tensor_descriptor.hpp"
#include "ck/tensor_description/tensor_descriptor_helper.hpp"
#include "ck/tensor_operation/gpu/device/gemm_specialization.hpp"

Go to the source code of this file.

Classes
struct	ck::tensor_operation::device::GemmGemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType, OPerTileType >
struct	ck::tensor_operation::device::GemmPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType >
struct	ck::tensor_operation::device::MatrixPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType >
struct	ck::tensor_operation::device::GemmPadder_v2< PadM, PadN, PadK, MPerTileType, NPerTileType, KPerTileType >
struct	ck::tensor_operation::device::MatrixPadder_v2< PadM, PadN, PadK, MPerTileType, NPerTileType, KPerTileType >

Namespaces
namespace	ck
namespace	ck::tensor_operation
namespace	ck::tensor_operation::device

Functions
template<typename TensorDesc, typename TileLengths, typename DoPads>
__host__ __device__ constexpr auto	ck::tensor_operation::device::PadTensorDescriptor (const TensorDesc &desc, const TileLengths &tile_lengths, DoPads)
template<GemmSpecialization GemmSpec, typename MPerTileType, typename NPerTileType, typename KPerTileType, typename CDesc_MRaw_NRaw>
auto	ck::tensor_operation::device::grid_desc (MatrixPadder< GemmSpec, MPerTileType, NPerTileType, KPerTileType > matrix_padder, CDesc_MRaw_NRaw conv_desc)