diff options
Diffstat (limited to 'ArmnnDriver.hpp')
-rw-r--r-- | ArmnnDriver.hpp | 22 |
1 files changed, 21 insertions, 1 deletions
diff --git a/ArmnnDriver.hpp b/ArmnnDriver.hpp index 7c6e5d0b..d961f861 100644 --- a/ArmnnDriver.hpp +++ b/ArmnnDriver.hpp @@ -9,7 +9,27 @@ #include <log/log.h> -#ifdef ARMNN_ANDROID_NN_V1_1 // Using ::android::hardware::neuralnetworks::V1_1 +#ifdef ARMNN_ANDROID_NN_V1_2 // Using ::android::hardware::neuralnetworks::V1_2 + +#include "1.1/ArmnnDriver.hpp" +#include "1.2/ArmnnDriver.hpp" + +namespace armnn_driver +{ + +class ArmnnDriver : public hal_1_2::ArmnnDriver +{ +public: + ArmnnDriver(DriverOptions options) + : hal_1_2::ArmnnDriver(std::move(options)) + { + ALOGV("ArmnnDriver::ArmnnDriver()"); + } + ~ArmnnDriver() {} +}; + +} // namespace armnn_driver +#elif ARMNN_ANDROID_NN_V1_1 // Using ::android::hardware::neuralnetworks::V1_1 #include "1.1/ArmnnDriver.hpp" |