diff options
| -rw-r--r-- | arm_compute/core/utils/misc/ShapeCalculator.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h index f5058b35fb..26b337d5c5 100644 --- a/arm_compute/core/utils/misc/ShapeCalculator.h +++ b/arm_compute/core/utils/misc/ShapeCalculator.h @@ -1204,6 +1204,7 @@ inline TensorShape calculate_concatenate_shape(const std::vector<T *> &input, si { TensorShape out_shape = extract_shape(input[0]); +#if defined(ARM_COMPUTE_ASSERTS_ENABLED) // All dimensions must match except the axis one for(unsigned int i = 0; i < MAX_DIMS; ++i) { @@ -1219,6 +1220,7 @@ inline TensorShape calculate_concatenate_shape(const std::vector<T *> &input, si ARM_COMPUTE_ERROR_ON(out_shape[i] != shape[i]); } } +#endif // defined(ARM_COMPUTE_ASSERTS_ENABLED) // Calculate output shape size_t new_size = 0; |