Typedef mindspore::registry::CreateKernel

• Defined in File register_kernel.h

Typedef Documentation

using mindspore::registry::CreateKernel = 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)>

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.