diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-09-02 09:55:14 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-09-02 09:55:14 +0100 |
commit | 346e8119e51cd9a3fbb56eb6fa85ad48a256c8eb (patch) | |
tree | afb19ec685e46421fab84918630251fd92083d9d /ConversionUtils.hpp | |
parent | 82609564981ca40d0fc86d2bb4b8ee62648a41c6 (diff) | |
download | android-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
Diffstat (limited to 'ConversionUtils.hpp')
-rw-r--r-- | ConversionUtils.hpp | 3 |
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__); } |