/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/library/utility/conv_common.hpp Source File#
conv_common.hpp
Go to the documentation of this file.
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)
Definition: conv_common.hpp:14
constexpr std::size_t calculate_convolution_flops(const InDesc &, const WeiDesc &wei_desc, const OutDesc &out_desc)
Definition: conv_common.hpp:58
Definition: ck.hpp:267
__host__ constexpr __device__ auto make_naive_tensor_descriptor_packed(const Tuple< Lengths... > &lengths)
Definition: tensor_descriptor_helper.hpp:101
Definition: tensor_descriptor.hpp:28
__host__ static constexpr __device__ index_t GetNumOfDimension()
Definition: tensor_descriptor.hpp:141
__host__ constexpr __device__ auto GetLength(Number< IDim >) const
Definition: tensor_descriptor.hpp:147
Definition: integral_constant.hpp:20