aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2020-09-02 09:55:14 +0100
committerSadik Armagan <sadik.armagan@arm.com>2020-09-02 09:55:14 +0100
commit346e8119e51cd9a3fbb56eb6fa85ad48a256c8eb (patch)
treeafb19ec685e46421fab84918630251fd92083d9d
parent82609564981ca40d0fc86d2bb4b8ee62648a41c6 (diff)
downloadandroid-nn-driver-346e8119e51cd9a3fbb56eb6fa85ad48a256c8eb.tar.gz
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
-rw-r--r--ConversionUtils.hpp3
1 files changed, 2 insertions, 1 deletions
diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp
index 830502df..cdb57d1f 100644
--- a/ConversionUtils.hpp
+++ b/ConversionUtils.hpp
@@ -3671,7 +3671,8 @@ bool ConvertSqueeze(const HalOperation& operation, const HalModel& model, Conver
{
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__);
}