ArmnnDriverImpl.hpp

Go to the documentation of this file.

//
// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
 
#pragma once
 
#include "DriverOptions.hpp"
 
#include <armnn/ArmNN.hpp>
 
#include <nnapi/IPreparedModel.h>
#include <nnapi/Result.h>
#include <nnapi/TypeUtils.h>
#include <nnapi/Types.h>
#include <nnapi/Validation.h>
 
using namespace android::nn;
 
namespace armnn_driver
{
 
class ArmnnDriverImpl
{
public:
    static GeneralResult<SharedPreparedModel> PrepareArmnnModel(
        const armnn::IRuntimePtr& runtime,
        const armnn::IGpuAccTunedParametersPtr& clTunedParameters,
        const DriverOptions& options,
        const Model& model,
        const std::vector<SharedHandle>& modelCacheHandle,
        const std::vector<SharedHandle>& dataCacheHandle,
        const CacheToken& token,
        bool float32ToFloat16 = false,
        Priority priority = Priority::MEDIUM);
 
    static GeneralResult<SharedPreparedModel> PrepareArmnnModelFromCache(
        const armnn::IRuntimePtr& runtime,
        const armnn::IGpuAccTunedParametersPtr& clTunedParameters,
        const DriverOptions& options,
        const std::vector<SharedHandle>& modelCacheHandle,
        const std::vector<SharedHandle>& dataCacheHandle,
        const CacheToken& token,
        bool float32ToFloat16 = false);
 
    static const Capabilities& GetCapabilities(const armnn::IRuntimePtr& runtime);
 
private:
    static bool ValidateSharedHandle(const SharedHandle& sharedHandle);
    static bool ValidateDataCacheHandle(const std::vector<SharedHandle>& dataCacheHandle, const size_t dataSize);
};
 
} // namespace armnn_driver