aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2020-05-22 08:34:16 +0100
committerJan Eilers <jan.eilers@arm.com>2020-05-22 12:34:27 +0100
commit5932294b2f3d305352c3e6dbbeeb5f23a4679023 (patch)
tree3fea016009274d427a450fb6ed3ae834947db3b6
parentd275cb2fcb3b41928348a408f7eace4ff516ca8b (diff)
downloadandroid-nn-driver-5932294b2f3d305352c3e6dbbeeb5f23a4679023.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.hpp4
1 files changed, 2 insertions, 2 deletions
diff --git a/ConversionUtils_1_3.hpp b/ConversionUtils_1_3.hpp
index 4714b84..dada670 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__);
}