diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-05-19 14:10:30 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-05-19 14:10:30 +0100 |
commit | 813f23049d73177edfc1f1cff71147c39f4b695e (patch) | |
tree | cef15224e1e7a4bd8117bdc9f57a71be84e4325d /1.2 | |
parent | abc95d04dfb2462ffb42bc1facde4f45ecc65319 (diff) | |
download | android-nn-driver-813f23049d73177edfc1f1cff71147c39f4b695e.tar.gz |
IVGCVSW-4453 Add Support for ANEURALNETWORKS_QLSTM to HAL 1.3 Driver
* Add QLSTM support for Android NN Driver
* Add overrideOutputInfo parameter to SetupAndTrackLayerOutputSlot
* Add optional condition to GetInputScalar
* Refactor Quantized 16 Bit LSTM impl
Change-Id: Ie8fa98ad5ee4a62174ef91ca80f1df62b7fde937
Signed-off-by: Keith Davis <keith.davis@arm.com>
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Diffstat (limited to '1.2')
-rw-r--r-- | 1.2/HalPolicy.cpp | 8 | ||||
-rw-r--r-- | 1.2/HalPolicy.hpp | 2 |
2 files changed, 5 insertions, 5 deletions
diff --git a/1.2/HalPolicy.cpp b/1.2/HalPolicy.cpp index 9e547fab..d55e587f 100644 --- a/1.2/HalPolicy.cpp +++ b/1.2/HalPolicy.cpp @@ -100,7 +100,7 @@ bool HalPolicy::ConvertOperation(const Operation& operation, const Model& model, case V1_2::OperationType::QUANTIZE: return ConvertQuantize(operation, model, data); case V1_2::OperationType::QUANTIZED_16BIT_LSTM: - return ConvertQuantizedLstm(operation, model, data); + return ConvertQuantized16BitLstm(operation, model, data); case V1_2::OperationType::RELU: return ConvertReLu(operation, model, data); case V1_2::OperationType::RELU1: @@ -338,10 +338,10 @@ bool HalPolicy::ConvertQuantize(const Operation& operation, const Model& model, return ::ConvertQuantize<hal_1_2::HalPolicy>(operation, model, data); } -bool HalPolicy::ConvertQuantizedLstm(const Operation& operation, const Model& model, ConversionData& data) +bool HalPolicy::ConvertQuantized16BitLstm(const Operation& operation, const Model& model, ConversionData& data) { - ALOGV("hal_1_2::HalPolicy::ConvertQuantizedLstm()"); - return ::ConvertQuantizedLstm<hal_1_2::HalPolicy>(operation, model, data); + ALOGV("hal_1_2::HalPolicy::ConvertQuantized16BitLstm()"); + return ::ConvertQuantized16BitLstm<hal_1_2::HalPolicy>(operation, model, data); } bool HalPolicy::ConvertReLu(const Operation& operation, const Model& model, ConversionData& data) diff --git a/1.2/HalPolicy.hpp b/1.2/HalPolicy.hpp index b127a638..ce43a6e5 100644 --- a/1.2/HalPolicy.hpp +++ b/1.2/HalPolicy.hpp @@ -110,7 +110,7 @@ private: static bool ConvertQuantize(const Operation& operation, const Model& model, ConversionData& data); - static bool ConvertQuantizedLstm(const Operation& operation, const Model& model, ConversionData& data); + static bool ConvertQuantized16BitLstm(const Operation& operation, const Model& model, ConversionData& data); static bool ConvertReLu(const Operation& operation, const Model& model, ConversionData& data); |