Class KernelInterfaceReg
Defined in File register_kernel_interface.h
Class Documentation
-
class KernelInterfaceReg
KernelInterfaceReg defined registration class of KernelInterface.
Public Functions
-
inline KernelInterfaceReg(const std::string &provider, int op_type, const KernelInterfaceCreator creator)
Constructor of KernelInterfaceReg to register an ordinary op.
- Parameters
provider – [in] Define the identification of user.
op_type – [in] Define the ordinary op type.
creator – [in] Define the KernelInterface create function.
-
inline KernelInterfaceReg(const std::string &provider, const std::string &op_type, const KernelInterfaceCreator creator)
Constructor of KernelInterfaceReg to register custom op.
- Parameters
provider – [in] Define the identification of user.
op_type – [in] Define the concrete type of a custom op.
creator – [in] Define the KernelInterface create function.
-
inline KernelInterfaceReg(const std::string &provider, int op_type, const KernelInterfaceCreator creator)
Constructor of KernelInterfaceReg to register an ordinary op.
- Parameters
provider – [in] Define the identification of user.
op_type – [in] Define the ordinary op type.
creator – [in] Define the KernelInterface create function.
-
inline KernelInterfaceReg(const std::string &provider, const std::string &op_type, const KernelInterfaceCreator creator)
Constructor of KernelInterfaceReg to register custom op.
- Parameters
provider – [in] Define the identification of user.
op_type – [in] Define the concrete type of a custom op.
creator – [in] Define the KernelInterface create function.
-
inline KernelInterfaceReg(const std::string &provider, int op_type, const KernelInterfaceCreator creator)