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

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

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/library/utility/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 >

constexpr 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 >

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
	)