aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/reference/FuseBatchNormalization.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/validation/reference/FuseBatchNormalization.cpp')
-rw-r--r--tests/validation/reference/FuseBatchNormalization.cpp6
1 files changed, 5 insertions, 1 deletions
diff --git a/tests/validation/reference/FuseBatchNormalization.cpp b/tests/validation/reference/FuseBatchNormalization.cpp
index df12b25912..cb5003874b 100644
--- a/tests/validation/reference/FuseBatchNormalization.cpp
+++ b/tests/validation/reference/FuseBatchNormalization.cpp
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2019 ARM Limited.
+ * Copyright (c) 2019-2020 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -22,6 +22,7 @@
* SOFTWARE.
*/
#include "FuseBatchNormalization.h"
+#include "tests/validation/Helpers.h"
namespace arm_compute
{
@@ -45,6 +46,9 @@ void fuse_batch_normalization_dwc_layer(const SimpleTensor<T> &w, const SimpleTe
const unsigned int height = w.shape()[1];
const unsigned int dim2 = w.shape()[2];
+#if defined(_OPENMP)
+ #pragma omp parallel for
+#endif /* _OPENMP */
for(unsigned int b = 0; b < dim2; ++b)
{
const auto mean_val = mean.data()[b];