device_grouped_conv_utils.hpp File Reference

device_grouped_conv_utils.hpp File Reference#

Composable Kernel: device_grouped_conv_utils.hpp File Reference
device_grouped_conv_utils.hpp File Reference

Go to the source code of this file.

Classes

struct  ck::tensor_operation::device::ComputePtrOffsetOfStridedBatch< NumATensor, NumBTensor, NumDTensor, typename >
struct  ck::tensor_operation::device::ComputePtrOffsetOfStridedBatch< NumATensor, NumBTensor, NumDTensor, enable_if_t<(NumATensor > 1||NumBTensor > 1)> >
struct  ck::tensor_operation::device::ComputePtrOffsetOfStridedBatch< NumATensor, NumBTensor, NumDTensor, enable_if_t<(NumATensor==1 &&NumBTensor==1)> >

Namespaces

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

Functions

template<typename InLayout, typename WeiLayout, typename OutLayout>
constexpr bool ck::tensor_operation::device::is_NWGC_GKXC_NWGK ()
template<typename InLayout, typename WeiLayout, typename OutLayout>
constexpr bool ck::tensor_operation::device::is_GNWC_GKXC_GNWK ()
template<typename InLayout, typename WeiLayout, typename OutLayout>
constexpr bool ck::tensor_operation::device::is_NGCW_GKXC_NGKW ()
template<typename InLayout, typename WeiLayout, typename OutLayout>
constexpr bool ck::tensor_operation::device::is_NHWGC_GKYXC_NHWGK ()
template<typename InLayout, typename WeiLayout, typename OutLayout>
constexpr bool ck::tensor_operation::device::is_GNHWC_GKYXC_GNHWK ()
template<typename InLayout, typename WeiLayout, typename OutLayout>
constexpr bool ck::tensor_operation::device::is_NGCHW_GKYXC_NGKHW ()
template<typename InLayout, typename WeiLayout, typename OutLayout>
constexpr bool ck::tensor_operation::device::is_NGCHW_GKCYX_NGKHW ()
template<typename InLayout, typename WeiLayout, typename OutLayout>
constexpr bool ck::tensor_operation::device::is_NGCHW_NGKHW ()
template<typename InLayout, typename WeiLayout, typename OutLayout>
constexpr bool ck::tensor_operation::device::is_NDHWGC_GKZYXC_NDHWGK ()
template<typename InLayout, typename WeiLayout, typename OutLayout>
constexpr bool ck::tensor_operation::device::is_GNDHWC_GKZYXC_GNDHWK ()
template<typename InLayout, typename WeiLayout, typename OutLayout>
constexpr bool ck::tensor_operation::device::is_NGCDHW_GKZYXC_NGKDHW ()
template<typename InLayout, typename WeiLayout, typename OutLayout>
constexpr bool ck::tensor_operation::device::is_NGCDHW_GKCZYX_NGKDHW ()
template<typename InLayout, typename WeiLayout, typename OutLayout>
constexpr bool ck::tensor_operation::device::is_NGCDHW_NGKDHW ()
template<typename InLayout, typename WeiLayout, typename OutLayout>
constexpr bool ck::tensor_operation::device::is_NSpatialGC_GKSpatial_NSpatialGK ()
template<typename InLayout, typename WeiLayout, typename OutLayout>
constexpr bool ck::tensor_operation::device::is_GNSpatialC_GKSpatial_GNSpatialK ()
template<typename InLayout, typename WeiLayout, typename OutLayout>
constexpr bool ck::tensor_operation::device::is_NGCSpatial_GKSpatial_NGKSpatial ()