diff options
Diffstat (limited to 'tests/validation/fixtures/ReduceMeanFixture.h')
| -rw-r--r-- | tests/validation/fixtures/ReduceMeanFixture.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/validation/fixtures/ReduceMeanFixture.h b/tests/validation/fixtures/ReduceMeanFixture.h index 39fdea59fc..5363d6ba53 100644 --- a/tests/validation/fixtures/ReduceMeanFixture.h +++ b/tests/validation/fixtures/ReduceMeanFixture.h @@ -127,9 +127,9 @@ protected: #ifdef ARM_COMPUTE_OPENCL_ENABLED is_opencl = std::is_same<CLTensor, TensorType>::value; // Round down to zero on opencl to match kernel -#endif /* ARM_COMPUTE_OPENCL_ENABLED */ - out = reference::reduction_operation<T, T>(i == 0 ? src : out, output_shape, axis[i], ReductionOperation::MEAN_SUM, quantization_info_output, is_opencl ? RoundingPolicy::TO_ZERO : RoundingPolicy::TO_NEAREST_UP); - +#endif /* ARM_COMPUTE_OPENCL_ENABLED */ + out = reference::reduction_operation<T, T>(i == 0 ? src : out, output_shape, axis[i], ReductionOperation::MEAN_SUM, data_type, quantization_info_output, + is_opencl ? RoundingPolicy::TO_ZERO : RoundingPolicy::TO_NEAREST_UP); } if(!keep_dims) |