Navigation
| Name | NNE |
| Type | Runtime |
| Location | /Engine/Source/Runtime/NNE/ |
| Module Build Rules | NNE.Build.cs |
Classes
| Name | Remarks |
|---|---|
| FSharedModelData | This class implements a reference counted view on an immutable memory buffer representing model data. |
| FSymbolicTensorShape | A symbolic tensor shape represents the shape of a tensor with potentially variable dimension. |
| FTensorDesc | The descriptor for a tensor as model inputs and output. |
| FTensorShape | The concrete shape of a tensor. |
| UNNEModelData | This class represents assets that store neural network model data. |
| UNNERuntime | |
| UNNERuntimeCPU | |
| UNNERuntimeGPU | |
| UNNERuntimeNPU | |
| UNNERuntimeRDG |
Structs
| Name | Remarks |
|---|---|
| FTensorBindingCPU | The tensor binding for passing input and output to from CPU memory. |
| FTensorBindingRDG | The tensor binding for passing input and output to IModelRDG. |
Interfaces
| Name | Remarks |
|---|---|
| IModelCPU | The interface of a model capable of creating model instance that can run on CPU. |
| IModelGPU | The interface of a model capable of creating model instance that can run on GPU. |
| IModelInstanceCPU | The interface of a model instance that can run on CPU. |
| IModelInstanceGPU | The interface of a model instance that can run on GPU. |
| IModelInstanceNPU | The interface of a model instance that can run on NPU. |
| IModelInstanceRDG | The interface of a model instance that can run on RDG. |
| IModelInstanceRunSync | The interface of a model instance that can run synchronously from CPU memory. |
| IModelNPU | The interface of a model capable of creating model instance that can run on NPU. |
| IModelRDG | The interface of a model capable of creating model instance that can run on RDG. |
| INNERuntime | The base interface of a neural network runtime. |
| INNERuntimeCPU | The interface of a neural network runtime capable of creating CPU models. |
| INNERuntimeGPU | The interface of a neural network runtime capable of creating GPU models. |
| INNERuntimeNPU | The interface of a neural network runtime capable of creating NPU models. |
| INNERuntimeRDG | The interface of a neural network runtime capable of creating RDG models. |
Typedefs
| Name | Type | Remarks | Include Path |
|---|---|---|---|
| ERegisterRuntimeStatus | EResultStatus | NNE.h | |
| EUnregisterRuntimeStatus | EResultStatus | NNE.h | |
| FRDGBufferRef | FRDGBuffer * | NNERuntimeRDG.h | |
| FTensorBindingGPU | FTensorBindingCPU | NNERuntimeGPU.h |
Enums
Public
| Name | Remarks |
|---|---|
| ENNETensorDataType | The enum lists all tensor data types used in NNE. |
| UE::NNE::EResultStatus |
Functions
Public
| Name | Remarks | Include Path | Unreal Specifiers |
|---|---|---|---|
TArray< FString > UE::NNE::GetAllRuntimeNames() |
List and return all registered runtime names that implement the provided interface. | NNE.h | |
TWeakInterfacePtr< T > UE::NNE::GetRuntime
(
const FString& Name |
Find and return a runtime by name and interface. | NNE.h |