跳至内容

UModelInstanceWrapper

template<class T>
class UModelInstanceWrapper : public IModelInstanceInterface

用于包装针对不同设备的 NNE 模型的通用接口。

模板参数: class T

依赖项: FGenericTensorBinding, IModelInstanceInterface

继承自: public IModelInstanceInterface

公共接口

构造函数

UModelInstanceWrapper

inline UModelInstanceWrapper(TSharedPtr<T> ModelPtr)

参数

名称类型默认
ModelPtrTSharedPtr<T>

属性: inline

来源: c:/Git/Schola/Source/Schola/Public/Policies/InferencePolicy.h (line 51, column 1)

公共函数

GetInputTensorDescs

inline virtual TConstArrayView<UE::NNE::FTensorDesc> GetInputTensorDescs()

返回: TConstArrayView&lt;UE::NNE::FTensorDesc&gt;

属性: 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

参数

名称类型默认
InInputShapesTConstArrayView<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

参数

名称类型默认
InInputBindingsTConstArrayView<FGenericTensorBinding>
InOutputBindingsTConstArrayView<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)

© . This site is unofficial and not affiliated with AMD.