From 4bd9a745df49bdf11e03f932af6eca6b61ddb0a1 Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Wed, 12 Aug 2020 12:58:50 +0100 Subject: IVGCVSW-5182 Update Convert functions to use ShapeInferenceMethod. 1/2. * ConvertToActivation * ConvertAdd * ConvertArgMinMax * ConvertConv2d * ConvertDepthToSpace * ConvertDepthwiseConv2d * ConvertDiv * ConvertFloor * ConvertFullyConnected * ConvertL2Normalization * ConvertLocalResponseNormalization * ConvertMean * ConvertMul * ConvertPad * ConvertReshape * ConvertSub * ConvertStridedSlice * ConvertTranspose * ConvertBatchToSpaceNd * ConvertSpaceToBatchNd * ConvertComparison_1_2 * ConvertConv2d_1_2 * ConvertDepthwiseConv2d_1_2 * ConvertElementwiseUnary * ConvertExpandDims * ConvertGather * ConvertGroupedConv2d * ConvertInstanceNormalization * ConvertLogSoftmax * ConvertMaximum * ConvertMinimum * ConvertPadV2 * ConvertPrelu * ConvertQuantize * ConvertResize * ConvertSpaceToDepth * ConvertSoftmax * ConvertTransposeConv2d Signed-off-by: Finn Williams Signed-off-by: Teresa Charlin Signed-off-by: Kevin May Change-Id: Idacf16e5eab56d83fce293570bbc89381ae056dc --- Utils.cpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'Utils.cpp') diff --git a/Utils.cpp b/Utils.cpp index db1b6e68..77575d70 100644 --- a/Utils.cpp +++ b/Utils.cpp @@ -577,6 +577,11 @@ bool IsDynamicTensor(const armnn::TensorInfo& tensorInfo) { return true; } + // Account for the usage of the TensorShape empty constructor + if (tensorInfo.GetNumDimensions() == 0) + { + return true; + } return !tensorInfo.GetShape().AreAllDimensionsSpecified(); } -- cgit v1.2.1