diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-05-22 08:34:16 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-05-22 08:34:16 +0100 |
commit | 24af8b29c578b8819663a4bdc0828323123c2718 (patch) | |
tree | 3fea016009274d427a450fb6ed3ae834947db3b6 | |
parent | c60d0fd2cb57f0af7ecc6dc1fc678a9d2a2e629a (diff) | |
download | android-nn-driver-24af8b29c578b8819663a4bdc0828323123c2718.tar.gz |
IVGCVSW-4453 Add Support for ANEURALNETWORKS_QLSTM to HAL 1.3 Driver
* Input 30 for QLSTM is zero point of the hidden state, ANEURALNETWORKS_INT32
* Input 31 for QLSTM is the scale of the hidden state, ANEURALNETWORKS_FLOAT32
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I1c6304fa4772895ce57286b58b2c3363c25c38e2
-rw-r--r-- | ConversionUtils_1_3.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ConversionUtils_1_3.hpp b/ConversionUtils_1_3.hpp index 4714b84b..dada6704 100644 --- a/ConversionUtils_1_3.hpp +++ b/ConversionUtils_1_3.hpp @@ -331,8 +331,8 @@ bool ConvertQuantizedLstm(const HalOperation& operation, const HalModel& model, !GetInputScalar<HalPolicy>(operation, 27, HalOperandType::FLOAT32, matMulForgetGate, model, data) || !GetInputScalar<HalPolicy>(operation, 28, HalOperandType::FLOAT32, matMulCellGate, model, data) || !GetInputScalar<HalPolicy>(operation, 29, HalOperandType::FLOAT32, matMulOutputGate, model, data) || - !GetInputScalar<HalPolicy>(operation, 30, HalOperandType::FLOAT32, projInputScale, model, data) || - !GetInputScalar<HalPolicy>(operation, 31, HalOperandType::FLOAT32, projInputZeroPoint, model, data)) + !GetInputScalar<HalPolicy>(operation, 30, HalOperandType::INT32, projInputZeroPoint, model, data) || + !GetInputScalar<HalPolicy>(operation, 31, HalOperandType::FLOAT32, projInputScale, model, data)) { return Fail("%s: Operation has invalid scalar inputs", __func__); } |