Navigation
API > API/Runtime > API/Runtime/NNE
This class represents assets that store neural network model data.
Neural network models typically consist of a graph of operations and corresponding parameters as e.g. weights. UNNEModelData assets store such model data as imported e.g. by the UNNEModelDataFactory class. An INNERuntime object retrieved by UE::NNE::GetRuntime
| Name | UNNEModelData |
| Type | class |
| Header File | /Engine/Source/Runtime/NNE/Public/NNEModelData.h |
| Include Path | #include "NNEModelData.h" |
Syntax
UCLASS (BlueprintType, Category="NNE", MinimalAPI)
class UNNEModelData : public UObject
Inheritance Hierarchy
- UObjectBase → UObjectBaseUtility → UObject → UNNEModelData
Variables
Protected
| Name | Type | Remarks | Include Path | Unreal Specifiers |
|---|---|---|---|---|
| AdditionalFileData | TMap< FString, TArray64< uint8 > > | Additional raw binary data of the imported model. | NNEModelData.h | |
| FileData | TArray64< uint8 > | The raw binary file data of the imported model. | NNEModelData.h | |
| FileId | FGuid | A Guid that uniquely identifies this model. This is used to cache optimized models in the editor. | NNEModelData.h | |
| FileType | FString | A string identifying the type of data inside this asset. | NNEModelData.h | |
| ModelData | TMap< FString, TSharedPtr< UE::NNE::FSharedModelData > > | The processed / optimized model data for the different runtimes. | NNEModelData.h | |
| TargetRuntimes | TArray< FString > | A list of string of the supported runtime, empty to support them all. | NNEModelData.h |
Functions
Public
| Name | Remarks | Include Path | Unreal Specifiers |
|---|---|---|---|
void ClearFileDataAndFileType () |
Clears the FileData and the FileType. | NNEModelData.h | |
void ClearModelData () |
Clears the ModelData. | NNEModelData.h | |
TConstArrayView64< uint8 > GetAdditionalFileData
(
const FString& Key |
Get read only access to AdditionalFileData. | NNEModelData.h | |
virtual void GetAssetRegistryTags
(
FAssetRegistryTagsContext Context |
NNEModelData.h | ||
TConstArrayView64< uint8 > GetFileData () |
Get read only access to FileData. | NNEModelData.h | |
FGuid GetFileId () |
Get the FGuid identifying the FileData. | NNEModelData.h | |
FString GetFileType () |
Get the type of data inside FileData. | NNEModelData.h | |
TSharedPtr< UE::NNE::FSharedModelData > GetModelData
(
const FString& RuntimeName |
Get the cached (editor) or cooked (game) optimized model data for a given runtime. | NNEModelData.h | |
TArrayView< const FString > GetTargetRuntimes () |
In editor: Get the target runtimes this model data will be cooked for. | NNEModelData.h | |
void Init
(
const FString& Type, |
Initialize the model data with a copy of the data inside Buffer. | NNEModelData.h | |
virtual void Serialize
(
FArchive& Ar |
NNEModelData.h | ||
void SetTargetRuntimes
(
TArrayView< const FString > RuntimeNames |
Set the target runtimes this model data will be cooked for. An empty list means all runtimes. | NNEModelData.h |