UE::NNE::IModelRDG Class Reference

#include <NNERuntimeRDG.h>

Public Member Functions
virtual	~IModelRDG ()=default
virtual TSharedPtr< UE::NNE::IModelInstanceRDG >	CreateModelInstanceRDG ()=0

Detailed Description

The interface of a model capable of creating model instance that can run on RDG.

Use UE::NNE::GetRuntime<INNERuntimeRDG>(RuntimeName) to get a runtime capable of creating RDG models.

Constructor & Destructor Documentation

~IModelRDG()

virtual UE::NNE::IModelRDG::~IModelRDG ( )

virtualdefault

Member Function Documentation

CreateModelInstanceRDG()

virtual TSharedPtr< UE::NNE::IModelInstanceRDG > UE::NNE::IModelRDG::CreateModelInstanceRDG ( )

pure virtual

Create a model instance for inference

The runtime have the opportunity to share the model weights among multiple IModelInstanceRDG created from an IModelRDG instance, however this is not mandatory. The caller can decide to convert the result into a shared pointer if required (e.g. if the model needs to be shared with an async task for evaluation).

Returns

A caller owned model representing the neural network instance created.

---

The documentation for this class was generated from the following file:

• Engine/Source/Runtime/NNE/Public/NNERuntimeRDG.h