跳至内容

IModelInstanceInterface

class IModelInstanceInterface

一个模型实例的通用接口,包装针对不同设备的 NNE 模型实例。

公共接口

析构函数

~IModelInstanceInterface

virtual ~IModelInstanceInterface()=default

属性: virtual

来源: Schola/Source/Schola/Public/Policies/InferencePolicy.h (第 30 行,第 9 列)

公共函数

GetInputTensorDescs

pure-virtual TConstArrayView<UE::NNE::FTensorDesc> GetInputTensorDescs()=0

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

属性: pure-virtual

来源: Schola/Source/Schola/Public/Policies/InferencePolicy.h (第 31 行,第 25 列)

SetInputTensorShapes

pure-virtual UE::NNE::EResultStatus SetInputTensorShapes(TConstArrayView<UE::NNE::FTensorShape> InInputShapes)=0

参数

名称类型默认
InInputShapesTConstArrayView<UE::NNE::FTensorShape>

返回: UE::NNE::EResultStatus

属性: pure-virtual

来源: Schola/Source/Schola/Public/Policies/InferencePolicy.h (第 32 行,第 33 列)

RunSync

pure-virtual UE::NNE::EResultStatus RunSync(TConstArrayView<FTensorBindingCPU> InInputBindings, TConstArrayView<FTensorBindingCPU> InOutputBinding)=0

参数

名称类型默认
InInputBindingsTConstArrayView<FTensorBindingCPU>
InOutputBindingTConstArrayView<FTensorBindingCPU>

返回: UE::NNE::EResultStatus

属性: pure-virtual

来源: Schola/Source/Schola/Public/Policies/InferencePolicy.h (第 33 行,第 33 列)

由以下代码使用: UModelInstanceWrapper

来源: Schola/Source/Schola/Public/Policies/InferencePolicy.h (第 27 行,第 1 列)

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