diff options
author | Ferran Balaguer <ferran.balaguer@arm.com> | 2019-07-30 10:16:13 +0100 |
---|---|---|
committer | Ferran Balaguer Arm <ferran.balaguer@arm.com> | 2019-07-30 11:41:24 +0000 |
commit | a4a629ad3a29df52fb47592d435d41180367ec5f (patch) | |
tree | c29eef6a127c6913c3876283e2c9d75772003043 | |
parent | 613b49cdd8f4e40460711a4d84ed42f980477a17 (diff) | |
download | android-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.cpp | 8 |
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()); |