Class KernelReg

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.