DeviceGroupedConvFwd< NDimSpatial, InLayout, WeiLayout, OutLayout, InDataType, WeiDataType, OutDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation > Struct Template Reference

DeviceGroupedConvFwd&lt; NDimSpatial, InLayout, WeiLayout, OutLayout, InDataType, WeiDataType, OutDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation &gt; Struct Template Reference#

Composable Kernel: ck::tensor_operation::device::DeviceGroupedConvFwd< NDimSpatial, InLayout, WeiLayout, OutLayout, InDataType, WeiDataType, OutDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation > Struct Template Reference
Public Member Functions | List of all members
ck::tensor_operation::device::DeviceGroupedConvFwd< NDimSpatial, InLayout, WeiLayout, OutLayout, InDataType, WeiDataType, OutDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation > Struct Template Referenceabstract

#include <device_grouped_conv_fwd.hpp>

Inheritance diagram for ck::tensor_operation::device::DeviceGroupedConvFwd< NDimSpatial, InLayout, WeiLayout, OutLayout, InDataType, WeiDataType, OutDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation >:
ck::tensor_operation::device::BaseOperator

Public Member Functions

virtual std::unique_ptr< BaseArgument > MakeArgumentPointer (const void *p_in, const void *p_wei, void *p_out, const std::array< index_t, NDimSpatial+3 > &in_g_n_c_wis_lengths, const std::array< index_t, NDimSpatial+3 > &in_g_n_c_wis_strides, const std::array< index_t, NDimSpatial+3 > &wei_g_k_c_xs_lengths, const std::array< index_t, NDimSpatial+3 > &wei_g_k_c_xs_strides, const std::array< index_t, NDimSpatial+3 > &out_g_n_k_wos_lengths, const std::array< index_t, NDimSpatial+3 > &out_g_n_k_wos_strides, const std::array< index_t, NDimSpatial > &conv_filter_strides, const std::array< index_t, NDimSpatial > &conv_filter_dilations, const std::array< index_t, NDimSpatial > &input_left_pads, const std::array< index_t, NDimSpatial > &input_right_pads, const InElementwiseOperation &in_element_op, const WeiElementwiseOperation &wei_element_op, const OutElementwiseOperation &out_element_op)=0
 
virtual std::unique_ptr< BaseInvoker > MakeInvokerPointer ()=0
 
- Public Member Functions inherited from ck::tensor_operation::device::BaseOperator
 BaseOperator ()=default
 
 BaseOperator (const BaseOperator &)=default
 
BaseOperator & operator= (const BaseOperator &)=default
 
virtual bool IsSupportedArgument (const BaseArgument *)
 
virtual std::string GetTypeString () const
 
virtual std::string GetTypeIdName () const
 
virtual std::optional< std::string > GetObjectName () const
 
virtual std::optional< std::string > GetTemplateInfo () const
 
virtual std::string GetTypeIdHashCode () const
 
virtual size_t GetWorkSpaceSize (const BaseArgument *) const
 
virtual void SetWorkSpacePointer (BaseArgument *p_arg, void *p_workspace, const StreamConfig &=StreamConfig{}) const
 
virtual ~BaseOperator ()
 

Member Function Documentation

◆ MakeArgumentPointer()

template<index_t NDimSpatial, typename InLayout , typename WeiLayout , typename OutLayout , typename InDataType , typename WeiDataType , typename OutDataType , typename InElementwiseOperation , typename WeiElementwiseOperation , typename OutElementwiseOperation >
virtual std::unique_ptr<BaseArgument> ck::tensor_operation::device::DeviceGroupedConvFwd< NDimSpatial, InLayout, WeiLayout, OutLayout, InDataType, WeiDataType, OutDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation >::MakeArgumentPointer ( const void *  p_in,
const void *  p_wei,
void *  p_out,
const std::array< index_t, NDimSpatial+3 > &  in_g_n_c_wis_lengths,
const std::array< index_t, NDimSpatial+3 > &  in_g_n_c_wis_strides,
const std::array< index_t, NDimSpatial+3 > &  wei_g_k_c_xs_lengths,
const std::array< index_t, NDimSpatial+3 > &  wei_g_k_c_xs_strides,
const std::array< index_t, NDimSpatial+3 > &  out_g_n_k_wos_lengths,
const std::array< index_t, NDimSpatial+3 > &  out_g_n_k_wos_strides,
const std::array< index_t, NDimSpatial > &  conv_filter_strides,
const std::array< index_t, NDimSpatial > &  conv_filter_dilations,
const std::array< index_t, NDimSpatial > &  input_left_pads,
const std::array< index_t, NDimSpatial > &  input_right_pads,
const InElementwiseOperation &  in_element_op,
const WeiElementwiseOperation &  wei_element_op,
const OutElementwiseOperation &  out_element_op 
)
pure virtual

Implemented in ck::tensor_operation::device::DeviceGroupedConvFwdDl_NHWC_KYXC_NHWK< NDimSpatial, ADataType, BDataType, CDataType, AccDataType, ALayout, BLayout, CLayout, AElementwiseOperation, BElementwiseOperation, CElementwiseOperation, ConvForwardSpecialization, GemmSpec, BlockSize, MPerBlock, NPerBlock, K0PerBlock, K1, M1PerThread, N1PerThread, KPerThread, M1N1ThreadClusterM1Xs, M1N1ThreadClusterN1Xs, ABlockTransferThreadSliceLengths_K0_M0_M1_K1, ABlockTransferThreadClusterLengths_K0_M0_M1_K1, ABlockTransferThreadClusterArrangeOrder, ABlockTransferSrcAccessOrder, ABlockTransferSrcVectorTensorLengths_K0_M0_M1_K1, ABlockTransferSrcVectorTensorContiguousDimOrder, ABlockTransferDstVectorTensorLengths_K0_M0_M1_K1, BBlockTransferThreadSliceLengths_K0_N0_N1_K1, BBlockTransferThreadClusterLengths_K0_N0_N1_K1, BBlockTransferThreadClusterArrangeOrder, BBlockTransferSrcAccessOrder, BBlockTransferSrcVectorTensorLengths_K0_N0_N1_K1, BBlockTransferSrcVectorTensorContiguousDimOrder, BBlockTransferDstVectorTensorLengths_K0_N0_N1_K1, CThreadTransferSrcDstAccessOrder, CThreadTransferSrcDstVectorDim, CThreadTransferDstScalarPerVector, >.

◆ MakeInvokerPointer()

template<index_t NDimSpatial, typename InLayout , typename WeiLayout , typename OutLayout , typename InDataType , typename WeiDataType , typename OutDataType , typename InElementwiseOperation , typename WeiElementwiseOperation , typename OutElementwiseOperation >
virtual std::unique_ptr<BaseInvoker> ck::tensor_operation::device::DeviceGroupedConvFwd< NDimSpatial, InLayout, WeiLayout, OutLayout, InDataType, WeiDataType, OutDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation >::MakeInvokerPointer ( )
pure virtual

Implemented in ck::tensor_operation::device::DeviceGroupedConvFwdDl_NHWC_KYXC_NHWK< NDimSpatial, ADataType, BDataType, CDataType, AccDataType, ALayout, BLayout, CLayout, AElementwiseOperation, BElementwiseOperation, CElementwiseOperation, ConvForwardSpecialization, GemmSpec, BlockSize, MPerBlock, NPerBlock, K0PerBlock, K1, M1PerThread, N1PerThread, KPerThread, M1N1ThreadClusterM1Xs, M1N1ThreadClusterN1Xs, ABlockTransferThreadSliceLengths_K0_M0_M1_K1, ABlockTransferThreadClusterLengths_K0_M0_M1_K1, ABlockTransferThreadClusterArrangeOrder, ABlockTransferSrcAccessOrder, ABlockTransferSrcVectorTensorLengths_K0_M0_M1_K1, ABlockTransferSrcVectorTensorContiguousDimOrder, ABlockTransferDstVectorTensorLengths_K0_M0_M1_K1, BBlockTransferThreadSliceLengths_K0_N0_N1_K1, BBlockTransferThreadClusterLengths_K0_N0_N1_K1, BBlockTransferThreadClusterArrangeOrder, BBlockTransferSrcAccessOrder, BBlockTransferSrcVectorTensorLengths_K0_N0_N1_K1, BBlockTransferSrcVectorTensorContiguousDimOrder, BBlockTransferDstVectorTensorLengths_K0_N0_N1_K1, CThreadTransferSrcDstAccessOrder, CThreadTransferSrcDstVectorDim, CThreadTransferDstScalarPerVector, >.

The documentation for this struct was generated from the following file:
  • /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-7.0.1/include/ck/tensor_operation/gpu/device/device_grouped_conv_fwd.hpp