Navigation
API > API/Runtime > API/Runtime/NNE
References
| Module | NNE |
| Header | /Engine/Source/Runtime/NNE/Public/NNERuntimeRDG.h |
| Include | #include "NNERuntimeRDG.h" |
Syntax
class IModelInstanceRDG
Remarks
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
Destructors
| Type | Name | Description | |
|---|---|---|---|
Functions
| Type | Name | Description | |
|---|---|---|---|
| EEnqueueRDGStatus | EnqueueRDG
(
FRDGBuilder& RDGBuilder, |
Enqueue the model graph to a FRDGBuilder. | |
| TConstArrayView< FTensorDesc > | Get the input tensor descriptions as defined by the model, potentially with variable dimensions. | ||
| TConstArrayView< FTensorShape > | Get the input shapes.SetInputTensorShapes must be called prior of running a model. | ||
| TConstArrayView< FTensorDesc > | Get the output tensor descriptions as defined by the model, potentially with variable dimensions. | ||
| TConstArrayView< FTensorShape > | Getters for outputs shapes if they were already resolved. | ||
| ESetInputTensorShapesStatus | SetInputTensorShapes
(
TConstArrayView< FTensorShape > InInputShapes |
Prepare the model to be run with the given input shape. |
Typedefs
| Name | Description |
|---|---|
| EEnqueueRDGStatus | |
| ESetInputTensorShapesStatus |