commit ad486e21e5870f41774f30825c270762e08ae71e
author Gian Marco Iodice <gianmarco.iodice@arm.com> Tue Aug 07 17:17:06 2018 +0100
committer Anthony Barbier <anthony.barbier@arm.com> Fri Nov 02 16:54:54 2018 +0000
tree 71585d987c47dc6bdf7a5de23506b7b2e2352ca1
parent c6aa49b6709edada24b1ab3bc1308e0974f9e057

COMPMID-1188 - Fixed files extensions in GraphUtils.h

Also fixed the calculation of num_elements in access_numpy_tensor

Change-Id: Ic1a394ff829746d7803b81360830bade63b6b82a
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/143132
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>

utils/GraphUtils.cpp

diff --git a/utils/GraphUtils.cpp b/utils/GraphUtils.cpp
index ef09d3d..573c757 100644
--- a/utils/GraphUtils.cpp
+++ b/utils/GraphUtils.cpp
@@ -149,7 +149,7 @@
 template <typename T>
 void NumPyAccessor::access_numpy_tensor(ITensor &tensor)
 {
-    const int num_elements          = tensor.info()->total_size();
+    const int num_elements          = tensor.info()->tensor_shape().total_size();
     int       num_mismatches        = utils::compare_tensor<T>(tensor, _npy_tensor);
     float     percentage_mismatches = static_cast<float>(num_mismatches) / num_elements;