diff options
Diffstat (limited to 'arm_compute/core/utils')
| -rw-r--r-- | arm_compute/core/utils/misc/ShapeCalculator.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h index b256e73146..9d36405041 100644 --- a/arm_compute/core/utils/misc/ShapeCalculator.h +++ b/arm_compute/core/utils/misc/ShapeCalculator.h @@ -1162,6 +1162,11 @@ inline TensorShape extract_shape(T *data) return data->info()->tensor_shape(); } +inline TensorShape extract_shape(const ITensorInfo *data) +{ + return data->tensor_shape(); +} + inline TensorShape extract_shape(ITensorInfo *data) { return data->tensor_shape(); |