From a4a629ad3a29df52fb47592d435d41180367ec5f Mon Sep 17 00:00:00 2001 From: Ferran Balaguer Date: Tue, 30 Jul 2019 10:16:13 +0100 Subject: IVGCVSW-3594 Add LSTM Dynamic output tensors check Signed-off-by: Ferran Balaguer Change-Id: I7c0d704b53caeac3a950c6957c4cab11686f6793 --- 1.2/HalPolicy.cpp | 8 ++++++++ 1 file changed, 8 insertions(+) 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()); -- cgit v1.2.1