FGPUModelWrapper

Wrapper class for GPU-based neural network models.

This class wraps NNE GPU models to provide a consistent interface for creating model instances and checking validity.

class FGPUModelWrapper : public IModelInterface

Methods

FGPUModelWrapper

inline FGPUModelWrapper(TSharedPtr<UE::NNE::IModelGPU> RawModel)

Constructor that wraps a raw GPU model.

Parameters

• RawModel (TSharedPtr<UE::NNE::IModelGPU>)

---

IsValid

inline virtual bool IsValid()

Checks if the wrapped model is valid.

---

CreateModelInstance

inline virtual TSharedPtr<IModelInstanceRunSync> CreateModelInstance()

Creates a GPU model instance for inference.

Source: Source/ScholaNNE/Public/NNEUtils/NNEWrappers.h