1 // 2 // Copyright © 2020 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include <HalInterfaces.h> 9 10 #include "../DriverOptions.hpp" 11 12 #include <armnn/ArmNN.hpp> 13 14 using namespace android::nn; 15 16 namespace V1_0 = ::android::hardware::neuralnetworks::V1_0; 17 namespace V1_2 = ::android::hardware::neuralnetworks::V1_2; 18 namespace V1_3 = ::android::hardware::neuralnetworks::V1_3; 19 20 namespace armnn_driver 21 { 22 namespace hal_1_3 23 { 24 25 class ArmnnDriverImpl 26 { 27 public: 28 static android::hardware::Return<V1_3::ErrorStatus> prepareArmnnModel_1_3(const armnn::IRuntimePtr& runtime, 29 const armnn::IGpuAccTunedParametersPtr& clTunedParameters, 30 const DriverOptions& options, 31 const V1_3::Model& model, 32 const android::sp<V1_3::IPreparedModelCallback>& cb, 33 bool float32ToFloat16 = false, 34 V1_3::Priority priority = V1_3::Priority::MEDIUM); 35 36 static android::hardware::Return<void> getCapabilities_1_3(const armnn::IRuntimePtr& runtime, 37 V1_3::IDevice::getCapabilities_1_3_cb cb); 38 }; 39 40 } // namespace hal_1_3 41 } // namespace armnn_driver 42