1 files changed, 2 insertions, 3 deletions
diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h
index 7a34b43028..0d07266403 100644
--- a/arm_compute/core/utils/misc/ShapeCalculator.h
+++ b/arm_compute/core/utils/misc/ShapeCalculator.h
@@ -1162,12 +1162,11 @@ inline TensorShape extract_shape(T *data)
     return data->info()->tensor_shape();
 }
 
-inline TensorShape extract_shape(const ITensorInfo *data)
+inline TensorShape extract_shape(ITensorInfo *data)
 {
     return data->tensor_shape();
 }
-
-inline TensorShape extract_shape(ITensorInfo *data)
+inline TensorShape extract_shape(const ITensorInfo *data)
 {
     return data->tensor_shape();
 }