Define REGISTER_CUSTOM_KERNEL_INTERFACE

Define Documentation

REGISTER_CUSTOM_KERNEL_INTERFACE(provider, op_type, creator)

Defined registering macro to register custom op, which called by user directly.

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.