22 lines
515 B
C++
22 lines
515 B
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
#pragma once
|
|
|
|
#include "CoreMinimal.h"
|
|
#include "NNERuntimeRDGUtilsModelOptimizerBase.h"
|
|
|
|
namespace UE::NNERuntimeRDGUtils::Private
|
|
{
|
|
|
|
class FModelOptimizerONNXToNNERT : public FModelOptimizerBase
|
|
{
|
|
public:
|
|
virtual FString GetName() const override
|
|
{
|
|
return TEXT("NNEModelOptimizerFromONNXToNNERuntimeRDGHlsl");
|
|
}
|
|
|
|
virtual bool Optimize(TConstArrayView<uint8> InputModel, TArray<uint8>& OutModel) override;
|
|
};
|
|
|
|
} // namespace UE::NNERuntimeRDGUtils::Private
|