aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorFerran Balaguer <ferran.balaguer@arm.com>2019-07-30 10:16:13 +0100
committerFerran Balaguer Arm <ferran.balaguer@arm.com>2019-07-30 11:41:24 +0000
commita4a629ad3a29df52fb47592d435d41180367ec5f (patch)
treec29eef6a127c6913c3876283e2c9d75772003043
parent613b49cdd8f4e40460711a4d84ed42f980477a17 (diff)
downloadandroid-nn-driver-a4a629ad3a29df52fb47592d435d41180367ec5f.tar.gz
IVGCVSW-3594 Add LSTM Dynamic output tensors check
Signed-off-by: Ferran Balaguer <ferran.balaguer@arm.com> Change-Id: I7c0d704b53caeac3a950c6957c4cab11686f6793
-rw-r--r--1.2/HalPolicy.cpp8
1 files changed, 8 insertions, 0 deletions
diff --git a/1.2/HalPolicy.cpp b/1.2/HalPolicy.cpp
index 64d808bc..fe571df2 100644
--- a/1.2/HalPolicy.cpp
+++ b/1.2/HalPolicy.cpp
@@ -1405,6 +1405,14 @@ bool HalPolicy::ConvertLstm(const Operation& operation, const Model& model, Conv
const armnn::TensorInfo& cellStateOutInfo = GetTensorInfoForOperand(*cellStateOut);
const armnn::TensorInfo& outputInfo = GetTensorInfoForOperand(*output);
+ if (IsDynamicTensor(scratchBufferInfo) ||
+ IsDynamicTensor(outputStateOutInfo) ||
+ IsDynamicTensor(cellStateOutInfo) ||
+ IsDynamicTensor(outputInfo))
+ {
+ return Fail("%s: Dynamic output tensors are not supported", __func__);
+ }
+
// Basic parameters
armnn::LstmInputParamsInfo paramsInfo;
paramsInfo.m_InputToForgetWeights = &(params.m_InputToForgetWeights->GetInfo());