Navigation
API > API/Runtime > API/Runtime/NNE
The interface of a model instance that can run on RDG.
Use UE::NNE::IModelRDG::CreateModelInstance() to get a model instance. Use UE::NNE::GetRuntime
| Name | IModelInstanceRDG |
| Type | class |
| Header File | /Engine/Source/Runtime/NNE/Public/NNERuntimeRDG.h |
| Include Path | #include "NNERuntimeRDG.h" |
Syntax
class IModelInstanceRDG
Destructors
| Name | Remarks | Include Path | Unreal Specifiers |
|---|---|---|---|
virtual ~IModelInstanceRDG() |
NNERuntimeRDG.h |
Typedefs
| Name | Type | Remarks | Include Path |
|---|---|---|---|
| EEnqueueRDGStatus | EResultStatus | NNERuntimeRDG.h | |
| ESetInputTensorShapesStatus | EResultStatus | NNERuntimeRDG.h |
Functions
Public
| Name | Remarks | Include Path | Unreal Specifiers |
|---|---|---|---|
EEnqueueRDGStatus EnqueueRDG
(
FRDGBuilder& RDGBuilder, |
Enqueue the model graph to a FRDGBuilder. | NNERuntimeRDG.h | |
TConstArrayView< FTensorDesc > GetInputTensorDescs() |
Get the input tensor descriptions as defined by the model, potentially with variable dimensions. | NNERuntimeRDG.h | |
TConstArrayView< FTensorShape > GetInputTensorShapes() |
Get the input shapes.SetInputTensorShapes must be called prior of running a model. | NNERuntimeRDG.h | |
TConstArrayView< FTensorDesc > GetOutputTensorDescs() |
Get the output tensor descriptions as defined by the model, potentially with variable dimensions. | NNERuntimeRDG.h | |
TConstArrayView< FTensorShape > GetOutputTensorShapes () |
Getters for outputs shapes if they were already resolved. | NNERuntimeRDG.h | |
ESetInputTensorShapesStatus SetInputTensorShapes
(
TConstArrayView< FTensorShape > InInputShapes |
Prepare the model to be run with the given input shape. | NNERuntimeRDG.h |