diff options
Diffstat (limited to '1.2/ArmnnDriverImpl.hpp')
| -rw-r--r-- | 1.2/ArmnnDriverImpl.hpp | 19 |
1 files changed, 13 insertions, 6 deletions
diff --git a/1.2/ArmnnDriverImpl.hpp b/1.2/ArmnnDriverImpl.hpp index b3c65079..73ba133e 100644 --- a/1.2/ArmnnDriverImpl.hpp +++ b/1.2/ArmnnDriverImpl.hpp @@ -11,6 +11,13 @@ #include <armnn/ArmNN.hpp> +#ifdef ARMNN_ANDROID_R +using namespace android::nn::hal; +#endif + +namespace V1_0 = ::android::hardware::neuralnetworks::V1_0; +namespace V1_2 = ::android::hardware::neuralnetworks::V1_2; + namespace armnn_driver { namespace hal_1_2 @@ -19,12 +26,12 @@ namespace hal_1_2 class ArmnnDriverImpl { public: - static Return<ErrorStatus> prepareArmnnModel_1_2(const armnn::IRuntimePtr& runtime, - const armnn::IGpuAccTunedParametersPtr& clTunedParameters, - const DriverOptions& options, - const V1_2::Model& model, - const android::sp<V1_2::IPreparedModelCallback>& cb, - bool float32ToFloat16 = false); + static Return<V1_0::ErrorStatus> prepareArmnnModel_1_2(const armnn::IRuntimePtr& runtime, + const armnn::IGpuAccTunedParametersPtr& clTunedParameters, + const DriverOptions& options, + const V1_2::Model& model, + const android::sp<V1_2::IPreparedModelCallback>& cb, + bool float32ToFloat16 = false); static Return<void> getCapabilities_1_2(const armnn::IRuntimePtr& runtime, V1_2::IDevice::getCapabilities_1_2_cb cb); |