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.

参数

• 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.

参数

• 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.