Class KernelInterfaceReg

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.