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.

参数

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

参数

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