NNERuntimeCPU.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "NNERuntimeRunSync.h"
#include "NNEStatus.h"
#include "UObject/Interface.h"
 
#include "NNERuntimeCPU.generated.h"
 
class UNNEModelData;
 
namespace UE::NNE
{
 
class IModelInstanceCPU : public IModelInstanceRunSync
{
};
 
class IModelCPU
{
public:
 
    virtual ~IModelCPU() = default;
 
    virtual TSharedPtr<UE::NNE::IModelInstanceCPU> CreateModelInstanceCPU() = 0;
};
 
} // UE::NNE
 
UINTERFACE(MinimalAPI)
class UNNERuntimeCPU : public UInterface
{
    GENERATED_BODY()
};
 
class INNERuntimeCPU
{
    GENERATED_BODY()
    
public:
 
    using ECanCreateModelCPUStatus = UE::NNE::EResultStatus;
 
    virtual ECanCreateModelCPUStatus CanCreateModelCPU(const TObjectPtr<UNNEModelData> ModelData) const = 0;
    
    virtual TSharedPtr<UE::NNE::IModelCPU> CreateModelCPU(const TObjectPtr<UNNEModelData> ModelData) = 0;
};