Class KernelInterface

• Defined in File kernel_interface.h

Class Documentation

class KernelInterface

KernelInterface defined customized op's interface, such as infershape, and so on.

Public Functions

virtual ~KernelInterface() = default

Destructor of KernelInterface.

inline virtual Status Infer(std::vector<mindspore::MSTensor> *inputs, std::vector<mindspore::MSTensor> *outputs, const schema::Primitive *primitive)

Method to infer customized op's output shape.

参数

• inputs – [in] Define the input tensors of op.

• outputs – [in] Define the output tensors of op.

• primitive – [in] Define the attributes of op.

返回

Status as a status identification of inferring.

inline virtual Status Infer(std::vector<mindspore::MSTensor> *inputs, std::vector<mindspore::MSTensor> *outputs, const schema::Primitive *primitive, const Kernel *kernel)

Method to infer customized op's output shape.

参数

• inputs – [in] Define the input tensors of op.

• outputs – [in] Define the output tensors of op.

• primitive – [in] Define the attributes of op.

• kernel – [in] Define the kernel of a certain op.

返回

Status as a status identification of inferring.