commit 346e8119e51cd9a3fbb56eb6fa85ad48a256c8eb
author Sadik Armagan <sadik.armagan@arm.com> Wed Sep 02 09:55:14 2020 +0100
committer Sadik Armagan <sadik.armagan@arm.com> Wed Sep 02 09:55:14 2020 +0100
tree afb19ec685e46421fab84918630251fd92083d9d
parent 82609564981ca40d0fc86d2bb4b8ee62648a41c6

IVGCVSW-5276 'Update ConvertSqueeze function to use ShapeInferenceMethod'

* Enabled dynamic type tensors on Squeeze operator in nn-driver.

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I2092cf6087da2821a07bd4743c4676d5afb35e66

ConversionUtils.hpp

diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp
index 830502d..cdb57d1 100644
--- a/ConversionUtils.hpp
+++ b/ConversionUtils.hpp
@@ -3671,7 +3671,8 @@
     {
         return Fail("%s: Could not read output 0", __func__);
     }
-    if (IsDynamicTensor(GetTensorInfoForOperand(*output)))
+
+    if (IsDynamicTensor(GetTensorInfoForOperand(*output)) && !(AreDynamicTensorsSupported()))
     {
         return Fail("%s: Dynamic output tensors are not supported", __func__);
     }