commit a4a629ad3a29df52fb47592d435d41180367ec5f
author Ferran Balaguer <ferran.balaguer@arm.com> Tue Jul 30 10:16:13 2019 +0100
committer Ferran Balaguer Arm <ferran.balaguer@arm.com> Tue Jul 30 11:41:24 2019 +0000
tree c29eef6a127c6913c3876283e2c9d75772003043
parent 613b49cdd8f4e40460711a4d84ed42f980477a17

IVGCVSW-3594 Add LSTM Dynamic output tensors check

Signed-off-by: Ferran Balaguer <ferran.balaguer@arm.com>
Change-Id: I7c0d704b53caeac3a950c6957c4cab11686f6793

1.2/HalPolicy.cpp

diff --git a/1.2/HalPolicy.cpp b/1.2/HalPolicy.cpp
index 64d808b..fe571df 100644
--- a/1.2/HalPolicy.cpp
+++ b/1.2/HalPolicy.cpp
@@ -1405,6 +1405,14 @@
     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());