diff options
Diffstat (limited to '1.0/ArmnnDriverImpl.hpp')
-rw-r--r-- | 1.0/ArmnnDriverImpl.hpp | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/1.0/ArmnnDriverImpl.hpp b/1.0/ArmnnDriverImpl.hpp new file mode 100644 index 00000000..2628682d --- /dev/null +++ b/1.0/ArmnnDriverImpl.hpp @@ -0,0 +1,41 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// See LICENSE file in the project root for full license information. +// + +#pragma once + +#include <HalInterfaces.h> + +#include "DriverOptions.hpp" + +#include <armnn/ArmNN.hpp> + +namespace armnn_driver +{ +namespace V1_0 +{ + +class ArmnnDriverImpl +{ +public: + static Return<void> getCapabilities( + const armnn::IRuntimePtr& runtime, + ::android::hardware::neuralnetworks::V1_0::IDevice::getCapabilities_cb cb); + static Return<void> getSupportedOperations( + const armnn::IRuntimePtr& runtime, + const DriverOptions& options, + const ::android::hardware::neuralnetworks::V1_0::Model& model, + ::android::hardware::neuralnetworks::V1_0::IDevice::getSupportedOperations_cb cb); + static Return<ErrorStatus> prepareModel( + const armnn::IRuntimePtr& runtime, + const armnn::IGpuAccTunedParametersPtr& clTunedParameters, + const DriverOptions& options, + const ::android::hardware::neuralnetworks::V1_0::Model& model, + const android::sp<IPreparedModelCallback>& cb, + bool float32ToFloat16 = false); + static Return<DeviceStatus> getStatus(); +}; + +} // namespace armnn_driver::V1_0 +} // namespace armnn_driver |