Function mindspore::registry::CreateKernel

• Defined in File register_kernel.h

Function Documentation

std::shared_ptr<kernel::Kernel> mindspore::registry::CreateKernel(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.

Parameters:

• inputs – [in] Define input tensors of kernel.

• outputs – [in] Define output tensors of kernel.

• primitive – [in] Define attributes of op.

• ctx – [in] Define for holding environment variables during runtime.

Returns:

Smart Pointer of kernel.