Class KernelReg
Defined in File register_kernel.h
Class Documentation
-
class KernelReg
KernelReg Defined registration class of kernel.
Public Functions
-
~KernelReg() = default
Destructor of KernelReg.
-
inline KernelReg(const std::string &arch, const std::string &provider, DataType data_type, int op_type, const CreateKernel creator)
Method to register ordinary op.
- Parameters
arch – [in] Define deviceType, such as CPU.
provider – [in] Define the identification of user.
data_type – [in] Define kernel’s input data type.
op_type – [in] Define the ordinary op type.
creator – [in] Define a function pointer to create a kernel.
-
inline KernelReg(const std::string &arch, const std::string &provider, DataType data_type, const std::string &op_type, const CreateKernel creator)
Method to register customized op.
- Parameters
arch – [in] Define deviceType, such as CPU.
provider – [in] Define the identification of user.
data_type – [in] Define kernel’s input data type.
op_type – [in] Define the concrete type of a custom op.
creator – [in] Define a function pointer to create a kernel.
-
~KernelReg() = default
Destructor of KernelReg.
-
inline KernelReg(const std::string &arch, const std::string &provider, DataType data_type, int op_type, const CreateKernel creator)
Method to register ordinary op.
- Parameters
arch – [in] Define deviceType, such as CPU.
provider – [in] Define the identification of user.
data_type – [in] Define kernel’s input data type.
op_type – [in] Define the ordinary op type.
creator – [in] Define a function pointer to create a kernel.
-
inline KernelReg(const std::string &arch, const std::string &provider, DataType data_type, const std::string &op_type, const CreateKernel creator)
Method to register customized op.
- Parameters
arch – [in] Define deviceType, such as CPU.
provider – [in] Define the identification of user.
data_type – [in] Define kernel’s input data type.
op_type – [in] Define the concrete type of a custom op.
creator – [in] Define a function pointer to create a kernel.
-
~KernelReg() = default