Class Delegate

• Defined in File delegate.h

Inheritance Relationships

Base Type

• public mindspore::IDelegate< LiteDelegateGraph, kernel::Kernel, kernel::Kernel >

Derived Type

• public mindspore::CoreMLDelegate (Class CoreMLDelegate)

Class Documentation

class Delegate : public mindspore::IDelegate<LiteDelegateGraph, kernel::Kernel, kernel::Kernel>

Subclassed by mindspore::CoreMLDelegate

Public Functions

virtual Status Init() = 0

Init delegate.

说明

Init will be called in Model::Build.

返回

Status. If Status is kLiteNotSupport, the program will return to the MindSpore Lite inner inference.

inline std::shared_ptr<kernel::Kernel> CreateKernel(const std::shared_ptr<kernel::Kernel> &node) override

Create kernel.

参数

node – [in] The kernel to be created.

返回

Created kernel.

inline bool IsDelegateNode(const std::shared_ptr<kernel::Kernel> &node) override

Check if the node is delegate node.

参数

node – [in] The kernel to verify.

返回

True if the node is delegate.

inline void ReplaceNodes(const std::shared_ptr<LiteDelegateGraph> &graph) override

Replace the nodes in model with delegate nodes, delegate will create kernels by its delegate nodes.

参数

graph – [in] The graph to be built.

virtual Status Build(LiteDelegateGraph *model) = 0

Build delegate graph for MindSpore model.

说明

Build will be called in Model::Build.

说明

deprecated, use ReplaceNodes and CreateKernel to build delegate model.

参数

model – [in] Define the delegate model to be built.