Invoker Struct Reference

• ck
• tensor_operation
• device
• DeviceConv3dFwdNaive_Input_N_Di_Hi_Wi_C_Weight_K_Z_Y_X_C_Output_N_Do_Ho_Wo_K
• Invoker

Public Types | Public Member Functions | List of all members

ck::tensor_operation::device::DeviceConv3dFwdNaive_Input_N_Di_Hi_Wi_C_Weight_K_Z_Y_X_C_Output_N_Do_Ho_Wo_K< InDataType, WeiDataType, OutDataType, AccDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation >::Invoker Struct Reference

#include <device_conv3d_fwd_naive_ndhwc_kzyxc_ndhwk.hpp>

Inheritance diagram for ck::tensor_operation::device::DeviceConv3dFwdNaive_Input_N_Di_Hi_Wi_C_Weight_K_Z_Y_X_C_Output_N_Do_Ho_Wo_K< InDataType, WeiDataType, OutDataType, AccDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation >::Invoker:

Public Types
using	Argument = DeviceOp::Argument

Public Member Functions
float	Run (const Argument &arg, const StreamConfig &stream_config=StreamConfig{})
float	Run (const BaseArgument *p_arg, const StreamConfig &stream_config=StreamConfig{}) override
Public Member Functions inherited from ck::tensor_operation::device::BaseInvoker
	BaseInvoker ()=default
	BaseInvoker (const BaseInvoker &)=default
BaseInvoker &	operator= (const BaseInvoker &)=default
virtual	~BaseInvoker ()

Member Typedef Documentation

◆ Argument

template<typename InDataType , typename WeiDataType , typename OutDataType , typename AccDataType , typename InElementwiseOperation , typename WeiElementwiseOperation , typename OutElementwiseOperation >

using ck::tensor_operation::device::DeviceConv3dFwdNaive_Input_N_Di_Hi_Wi_C_Weight_K_Z_Y_X_C_Output_N_Do_Ho_Wo_K< InDataType, WeiDataType, OutDataType, AccDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation >::Invoker::Argument = DeviceOp::Argument

Member Function Documentation

◆ Run() [1/2]

template<typename InDataType , typename WeiDataType , typename OutDataType , typename AccDataType , typename InElementwiseOperation , typename WeiElementwiseOperation , typename OutElementwiseOperation >

float ck::tensor_operation::device::DeviceConv3dFwdNaive_Input_N_Di_Hi_Wi_C_Weight_K_Z_Y_X_C_Output_N_Do_Ho_Wo_K< InDataType, WeiDataType, OutDataType, AccDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation >::Invoker::Run ( const Argument &  arg, const StreamConfig &  stream_config = StreamConfig{}  )

inline

◆ Run() [2/2]

template<typename InDataType , typename WeiDataType , typename OutDataType , typename AccDataType , typename InElementwiseOperation , typename WeiElementwiseOperation , typename OutElementwiseOperation >

float ck::tensor_operation::device::DeviceConv3dFwdNaive_Input_N_Di_Hi_Wi_C_Weight_K_Z_Y_X_C_Output_N_Do_Ho_Wo_K< InDataType, WeiDataType, OutDataType, AccDataType, InElementwiseOperation, WeiElementwiseOperation, OutElementwiseOperation >::Invoker::Run ( const BaseArgument *  p_arg, const StreamConfig &  stream_config = StreamConfig{}  )

inlineoverridevirtual

Reimplemented from ck::tensor_operation::device::BaseInvoker.

---

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/impl/device_conv3d_fwd_naive_ndhwc_kzyxc_ndhwk.hpp