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.