Template Class DelegateModel

Class Documentation

template<class T>
class DelegateModel

Public Functions

inline DelegateModel(std::vector<kernel::Kernel*> *kernels, const std::vector<MSTensor> &inputs, const std::vector<MSTensor> &outputs, const std::map<kernel::Kernel*, const T*> &primitives, SchemaVersion version)

Constructor of MindSpore Lite DelegateModel.

~DelegateModel() = default

Destructor of MindSpore Lite DelegateModel.

inline const T *GetPrimitive(kernel::Kernel *kernel) const

Get Primitive of kernel::Kernel.

参数

kernel[in] kernel in DelegateModel kernels vector.

返回

The Primitive of The kernel.

inline KernelIter BeginKernelIterator()

Get the begin iterator of the DelegateModel kernels vector.

返回

The begin iterator of the DelegateModel kernels vector.

inline KernelIter EndKernelIterator()

Get the end iterator of the DelegateModel kernels vector.

返回

The end iterator of the DelegateModel kernels vector.

inline KernelIter Replace(KernelIter from, KernelIter end, kernel::Kernel *graph_kernel)

Replace the continuous kernel supported by the delegate with a delegate graph kernel.

参数
  • from[in] Define the begin iterator of continuous kernel supported by the delegate.

  • end[in] Define the end iterator of continuous kernel supported by the delegate.

返回

The next iterator after graph_kernel, point to the next kernel that is not visited.

inline std::vector<kernel::Kernel*> *nodes()

Get the nodes of DelegateModel.

返回

The pointer to nodes vector of DelegateModel.

inline const std::vector<mindspore::MSTensor> &inputs()

Get the input tensors of DelegateModel.

返回

The input tensor vector of DelegateModel.

inline const std::vector<mindspore::MSTensor> &outputs()

Get the output tensors of DelegateModel.

返回

The ioutput tensor vector of DelegateModel.

inline SchemaVersion GetVersion() const

Get the ms model version.

返回

The schema version for the primitives map.