diff options
author | Nikhil Raj <nikhil.raj@arm.com> | 2021-04-02 10:02:16 +0100 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2021-04-02 16:31:23 +0100 |
commit | c5e0bb093d24392ddd12c4bd6b6fe6d0d0de850d (patch) | |
tree | c8820953a3f5cc18b10018c862791b1f66eed76c | |
parent | ccfeb5ea9d3aa17f66c1c13f1682ac519c8973c0 (diff) | |
download | android-nn-driver-c5e0bb093d24392ddd12c4bd6b6fe6d0d0de850d.tar.gz |
IVGCVSW-5355 Skipped VTS DynamicOutputShape Expand Dims should be passing
Signed-off-by: Nikhil Raj <nikhil.raj@arm.com>
Change-Id: If6ed4e513cc0e8ee29fb76127d66893f8b98e30b
-rw-r--r-- | ConversionUtils_1_2.hpp | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/ConversionUtils_1_2.hpp b/ConversionUtils_1_2.hpp index 080b8264..cddb13d7 100644 --- a/ConversionUtils_1_2.hpp +++ b/ConversionUtils_1_2.hpp @@ -653,11 +653,6 @@ bool ConvertExpandDims(const HalOperation& operation, const HalModel& model, Con return Fail("%s: %s", __func__, e.what()); } - if (targetShape != outputInfo.GetShape()) - { - return Fail("%s: Shape of the output operand does not match the resolved expanded shape", __func__); - } - ReshapeDescriptor reshapeDescriptor; reshapeDescriptor.m_TargetShape = targetShape; @@ -675,6 +670,10 @@ bool ConvertExpandDims(const HalOperation& operation, const HalModel& model, Con if(!IsDynamicTensor(outputInfo)) { + if (targetShape != outputInfo.GetShape()) + { + return Fail("%s: Shape of the output operand does not match the resolved expanded shape", __func__); + } validateFunc(outputInfo, isSupported); } else |