Typedef mindspore::registry::CreateKernel
Defined in File register_kernel.h
Typedef Documentation
-
typedef std::function<std::shared_ptr<kernel::Kernel>(const std::vector<MSTensor> &inputs, const std::vector<MSTensor> &outputs, const schema::Primitive *primitive, const mindspore::Context *ctx)> mindspore::registry::CreateKernel
CreateKernel Defined a functor to create a kernel.
- Param inputs
[in] Define input tensors of kernel.
- Param outputs
[in] Define output tensors of kernel.
- Param primitive
[in] Define attributes of op.
- Param ctx
[in] Define for holding environment variables during runtime.
- Return
Smart Pointer of kernel.