Template Class IDelegate

std::vector<mindspore::MSTensor> outputs_

#include <delegate_api.h>

IDelegate定义了MindSpore Lite 创建Delegate（模板类）。

构造函数

IDelegate()

IDelegate(const std::vector<mindspore::MSTensor> &inputs, const std::vector<mindspore::MSTensor> &outputs)
      : AbstractDelegate(inputs, outputs)

析构函数

virtual ~IDelegate() = default

公有成员函数

ReplaceNodes

virtual void ReplaceNodes(const std::shared_ptr<Graph> &graph) = 0

替换Delegate节点。

IsDelegateNode

virtual bool IsDelegateNode(const std::shared_ptr<Node> &node) = 0

判断节点是否属于Delegate。

CreateKernel

virtual std::shared_ptr<Kernel> CreateKernel(const std::shared_ptr<Node> &node) = 0

创建Kernel。