conv_common.hpp File Reference

conv_common.hpp File Reference#

Composable Kernel: conv_common.hpp File Reference
Functions
conv_common.hpp File Reference
#include "ck/tensor_description/tensor_descriptor.hpp"

Go to the source code of this file.

Functions

template<typename... InDesc, typename... WeiDesc, typename ConvStrides, typename ConvDilations, typename LeftPads, typename RightPads>
constexpr auto get_convolution_output_default_4d_tensor_descriptor (const ck::TensorDescriptor< InDesc... > &in_desc, const ck::TensorDescriptor< WeiDesc... > &wei_desc, const ConvStrides &conv_strides, const ConvDilations conv_dilations, const LeftPads &left_pads, const RightPads &right_pads)
template<class InDesc, class WeiDesc, class OutDesc>
constexpr std::size_t calculate_convolution_flops (const InDesc &, const WeiDesc &wei_desc, const OutDesc &out_desc)

Function Documentation

◆ calculate_convolution_flops()

template<class InDesc, class WeiDesc, class OutDesc>
std::size_t calculate_convolution_flops ( const InDesc & ,
const WeiDesc & wei_desc,
const OutDesc & out_desc )
constexpr

◆ get_convolution_output_default_4d_tensor_descriptor()

template<typename... InDesc, typename... WeiDesc, typename ConvStrides, typename ConvDilations, typename LeftPads, typename RightPads>
auto get_convolution_output_default_4d_tensor_descriptor ( const ck::TensorDescriptor< InDesc... > & in_desc,
const ck::TensorDescriptor< WeiDesc... > & wei_desc,
const ConvStrides & conv_strides,
const ConvDilations conv_dilations,
const LeftPads & left_pads,
const RightPads & right_pads )
constexpr