aboutsummaryrefslogtreecommitdiff
path: root/Utils.cpp
diff options
context:
space:
mode:
authorAron Virginas-Tar <Aron.Virginas-Tar@arm.com>2019-07-23 14:01:37 +0100
committerAron Virginas-Tar <Aron.Virginas-Tar@arm.com>2019-07-23 14:33:20 +0100
commit573a8fa04ef0774fe44a4ac6669582a6bf753d6e (patch)
treeefb5f466c71af293597506ee77dcf9f944c6559f /Utils.cpp
parent074c25a1535b648fdf19d7f6648e8aceef9aa7ad (diff)
downloadandroid-nn-driver-573a8fa04ef0774fe44a4ac6669582a6bf753d6e.tar.gz
IVGCVSW-3553 Fix failing zero_sized tests
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Change-Id: Idd10f34babc0d2552d599872b853ba5fb5c98351
Diffstat (limited to 'Utils.cpp')
-rw-r--r--Utils.cpp6
1 files changed, 6 insertions, 0 deletions
diff --git a/Utils.cpp b/Utils.cpp
index c3c6310b..d3d62a02 100644
--- a/Utils.cpp
+++ b/Utils.cpp
@@ -328,4 +328,10 @@ void DumpJsonProfilingIfRequired(bool gpuProfilingEnabled,
profiler->Print(fileStream);
}
+bool IsDynamicTensor(const armnn::TensorInfo& outputInfo)
+{
+ // Dynamic tensors have at least one 0-sized dimension
+ return outputInfo.GetNumElements() == 0u;
+}
+
} // namespace armnn_driver