/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_utils.hpp Source File#
device_grouped_conv_utils.hpp
Go to the documentation of this file.
142 template <index_t NumATensor = 1, index_t NumBTensor = 1, index_t NumDTensor = 0, typename = void>
Array< ck::index_t, NumDTensor > BatchStrideDs_
Definition: device_grouped_conv_fwd_multiple_d_multiple_r_xdl_cshuffle.hpp:84
index_t BatchStrideE_
Definition: device_grouped_conv_fwd_multiple_d_multiple_r_xdl_cshuffle.hpp:85
constexpr bool is_NWGC_GKXC_NWGK()
Definition: device_grouped_conv_utils.hpp:15
constexpr bool is_NSpatialGC_GKSpatial_NSpatialGK()
Definition: device_grouped_conv_utils.hpp:119
constexpr bool is_GNWC_GKXC_GNWK()
Definition: device_grouped_conv_utils.hpp:23
constexpr bool is_GNDHWC_GKZYXC_GNDHWK()
Definition: device_grouped_conv_utils.hpp:88
constexpr bool is_NGCSpatial_GKSpatial_NGKSpatial()
Definition: device_grouped_conv_utils.hpp:135
constexpr bool is_NHWGC_GKYXC_NHWGK()
Definition: device_grouped_conv_utils.hpp:40
constexpr bool is_NGCHW_GKYXC_NGKHW()
Definition: device_grouped_conv_utils.hpp:56
constexpr bool is_NDHWGC_GKZYXC_NDHWGK()
Definition: device_grouped_conv_utils.hpp:80
constexpr bool is_NGCDHW_NGKDHW()
Definition: device_grouped_conv_utils.hpp:112
constexpr bool is_NGCW_GKXC_NGKW()
Definition: device_grouped_conv_utils.hpp:31
constexpr bool is_NGCHW_GKCYX_NGKHW()
Definition: device_grouped_conv_utils.hpp:64
constexpr bool is_GNSpatialC_GKSpatial_GNSpatialK()
Definition: device_grouped_conv_utils.hpp:127
constexpr bool is_NGCDHW_GKZYXC_NGKDHW()
Definition: device_grouped_conv_utils.hpp:96
constexpr bool is_GNHWC_GKYXC_GNHWK()
Definition: device_grouped_conv_utils.hpp:48
constexpr bool is_NGCDHW_GKCZYX_NGKDHW()
Definition: device_grouped_conv_utils.hpp:104
constexpr bool is_NGCHW_NGKHW()
Definition: device_grouped_conv_utils.hpp:72
Definition: ck.hpp:267
typename tuple_element< I, TTuple >::type tuple_element_t
Definition: tuple.hpp:208
Definition: integral_constant.hpp:20
Definition: functional2.hpp:33
__host__ constexpr __device__ long_index_t GetEPtrOffset(index_t g_idx) const
Definition: device_grouped_conv_utils.hpp:245
Array< long_index_t, NumDTensor > BatchStrideDs_
Definition: device_grouped_conv_utils.hpp:258
long_index_t BatchStrideA_
Definition: device_grouped_conv_utils.hpp:256
__host__ constexpr __device__ long_index_t GetBPtrOffset(index_t g_idx) const
Definition: device_grouped_conv_utils.hpp:232
long_index_t BatchStrideB_
Definition: device_grouped_conv_utils.hpp:257
long_index_t BatchStrideE_
Definition: device_grouped_conv_utils.hpp:259
__host__ constexpr __device__ long_index_t GetAPtrOffset(index_t g_idx) const
Definition: device_grouped_conv_utils.hpp:227
__host__ constexpr __device__ long_index_t GetCPtrOffset(index_t g_idx) const
Definition: device_grouped_conv_utils.hpp:251
ComputePtrOffsetOfStridedBatch()=default
ComputePtrOffsetOfStridedBatch(long_index_t BatchStrideA, long_index_t BatchStrideB, Array< long_index_t, NumDTensor > BatchStrideDs, long_index_t BatchStrideE)
Definition: device_grouped_conv_utils.hpp:216
__host__ constexpr __device__ auto GetDsPtrOffset(index_t g_idx) const
Definition: device_grouped_conv_utils.hpp:237
ComputePtrOffsetOfStridedBatch(Array< long_index_t, NumATensor > &BatchStrideAs, Array< long_index_t, NumBTensor > &BatchStrideBs, Array< long_index_t, NumDTensor > &BatchStrideDs, long_index_t BatchStrideE)
Definition: device_grouped_conv_utils.hpp:155
__host__ constexpr __device__ long_index_t GetEPtrOffset(index_t g_idx) const
Definition: device_grouped_conv_utils.hpp:190
ComputePtrOffsetOfStridedBatch()=default
__host__ constexpr __device__ long_index_t GetCPtrOffset(index_t g_idx) const
Definition: device_grouped_conv_utils.hpp:196
__host__ constexpr __device__ auto GetAsPtrOffset(index_t g_idx) const
Definition: device_grouped_conv_utils.hpp:166
Array< long_index_t, NumATensor > BatchStrideA_
Definition: device_grouped_conv_utils.hpp:201
Array< long_index_t, NumBTensor > BatchStrideB_
Definition: device_grouped_conv_utils.hpp:202
Array< long_index_t, NumDTensor > BatchStrideDs_
Definition: device_grouped_conv_utils.hpp:203
__host__ constexpr __device__ auto GetBsPtrOffset(index_t g_idx) const
Definition: device_grouped_conv_utils.hpp:174
long_index_t BatchStrideE_
Definition: device_grouped_conv_utils.hpp:204
__host__ constexpr __device__ auto GetDsPtrOffset(index_t g_idx) const
Definition: device_grouped_conv_utils.hpp:182