Skip to content

Class FGPUModelWrapper

Class FGPUModelWrapper

  • Defined in File NNEWrappers.h

Inheritance Relationships

Base Type

class FGPUModelWrapper : public IModelInterface

Source: Source/Schola/NNE/Public/NNEUtils/NNEWrappers.h 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.


Public Functions

SymbolDetails
FGPUModelWrapperConstructor that wraps a raw GPU model.
IsValidChecks if the wrapped model is valid.
CreateModelInstanceCreates a GPU model instance for inference.

FGPUModelWrapper

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

Constructor that wraps a raw GPU model.

Parameters

RawModel – [in] The NNE GPU model to wrap

#DirectionNameTypeDescription
1RawModelTSharedPtr< UE::NNE::IModelGPU >The NNE GPU model to wrap

Attributes: inline

Source: Source/Schola/NNE/Public/NNEUtils/NNEWrappers.h

IsValid

inline virtual bool IsValid()

Checks if the wrapped model is valid.

Returns:

true if the model pointer is valid, false otherwise

Attributes: inline, virtual

Source: Source/Schola/NNE/Public/NNEUtils/NNEWrappers.h

CreateModelInstance

inline virtual TSharedPtr<IModelInstanceRunSync> CreateModelInstance()

Creates a GPU model instance for inference.

Returns:

Shared pointer to the created GPU model instance

Attributes: inline, virtual

Source: Source/Schola/NNE/Public/NNEUtils/NNEWrappers.h