diff options
Diffstat (limited to 'tests/validation/reference/DepthConvertLayer.cpp')
-rw-r--r-- | tests/validation/reference/DepthConvertLayer.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tests/validation/reference/DepthConvertLayer.cpp b/tests/validation/reference/DepthConvertLayer.cpp index 57eeb7f6f3..9c6a9aa76c 100644 --- a/tests/validation/reference/DepthConvertLayer.cpp +++ b/tests/validation/reference/DepthConvertLayer.cpp @@ -46,6 +46,9 @@ SimpleTensor<T2> depth_convert(const SimpleTensor<T1> &src, DataType dt_out, Con // Up-casting if(element_size_from_data_type(src.data_type()) < element_size_from_data_type(dt_out)) { +#if defined(_OPENMP) + #pragma omp parallel for +#endif /* _OPENMP */ for(int i = 0; i < src.num_elements(); ++i) { result[i] = src[i] << shift; @@ -54,6 +57,9 @@ SimpleTensor<T2> depth_convert(const SimpleTensor<T1> &src, DataType dt_out, Con // Down-casting else { +#if defined(_OPENMP) + #pragma omp parallel for +#endif /* _OPENMP */ for(int i = 0; i < src.num_elements(); ++i) { T1 val = src[i] >> shift; |