aboutsummaryrefslogtreecommitdiff
path: root/Utils.cpp
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2020-08-12 12:58:50 +0100
committerFinn Williams <Finn.Williams@arm.com>2020-08-14 19:35:43 +0100
commit4bd9a745df49bdf11e03f932af6eca6b61ddb0a1 (patch)
treed6b596ed4d5d9bc0172ebdb4a60174fbbbec867c /Utils.cpp
parenta4983cec09a3e24bf4e99abd31aa11842e8b365f (diff)
downloadandroid-nn-driver-4bd9a745df49bdf11e03f932af6eca6b61ddb0a1.tar.gz
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 <Finn.Williams@Arm.com> Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com> Signed-off-by: Kevin May <kevin.may@arm.com> Change-Id: Idacf16e5eab56d83fce293570bbc89381ae056dc
Diffstat (limited to 'Utils.cpp')
-rw-r--r--Utils.cpp5
1 files changed, 5 insertions, 0 deletions
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();
}