UModelInstanceWrapper
template<class T>class UModelInstanceWrapper : public IModelInstanceInterface用于包装针对不同设备的 NNE 模型的通用接口。
模板参数: class T
依赖项: FGenericTensorBinding, IModelInstanceInterface
继承自: public IModelInstanceInterface
公共接口
构造函数
UModelInstanceWrapper
inline UModelInstanceWrapper(TSharedPtr<T> ModelPtr)参数
| 名称 | 类型 | 默认 |
|---|---|---|
ModelPtr | TSharedPtr<T> | “ |
属性: inline
来源: c:/Git/Schola/Source/Schola/Public/Policies/InferencePolicy.h (line 51, column 1)
公共函数
GetInputTensorDescs
inline virtual TConstArrayView<UE::NNE::FTensorDesc> GetInputTensorDescs()返回: TConstArrayView<UE::NNE::FTensorDesc>
属性: inline, virtual
来源: c:/Git/Schola/Source/Schola/Public/Policies/InferencePolicy.h (line 58, column 17)
SetInputTensorShapes
inline virtual UE::NNE::EResultStatus SetInputTensorShapes(TConstArrayView<UE::NNE::FTensorShape> InInputShapes) override参数
| 名称 | 类型 | 默认 |
|---|---|---|
InInputShapes | TConstArrayView<UE::NNE::FTensorShape> | “ |
返回: UE::NNE::EResultStatus
属性: inline, virtual
来源: c:/Git/Schola/Source/Schola/Public/Policies/InferencePolicy.h (line 63, column 24)
RunSync
inline virtual UE::NNE::EResultStatus RunSync(TConstArrayView<FGenericTensorBinding> InInputBindings, TConstArrayView<FGenericTensorBinding> InOutputBindings) override参数
| 名称 | 类型 | 默认 |
|---|---|---|
InInputBindings | TConstArrayView<FGenericTensorBinding> | “ |
InOutputBindings | TConstArrayView<FGenericTensorBinding> | “ |
返回: UE::NNE::EResultStatus
属性: inline, virtual
来源: c:/Git/Schola/Source/Schola/Public/Policies/InferencePolicy.h (line 68, column 24)
私有接口
私有成员
TSharedPtr<T> WrappedModel
TSharedPtr<T> WrappedModel属性: private
来源: c:/Git/Schola/Source/Schola/Public/Policies/InferencePolicy.h (line 40, column 12)
来源: c:/Git/Schola/Source/Schola/Public/Policies/InferencePolicy.h (line 37, column 1)