diff options
Diffstat (limited to 'tests/validation/fixtures/ReduceMeanFixture.h')
-rw-r--r-- | tests/validation/fixtures/ReduceMeanFixture.h | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/tests/validation/fixtures/ReduceMeanFixture.h b/tests/validation/fixtures/ReduceMeanFixture.h index 6debd4a038..8692213641 100644 --- a/tests/validation/fixtures/ReduceMeanFixture.h +++ b/tests/validation/fixtures/ReduceMeanFixture.h @@ -32,6 +32,7 @@ #include "tests/IAccessor.h" #include "tests/framework/Asserts.h" #include "tests/framework/Fixture.h" +#include "tests/validation/Helpers.h" #include "tests/validation/reference/ReductionOperation.h" #include "tests/validation/reference/ReshapeLayer.h" @@ -63,10 +64,8 @@ protected: } else { - const QuantizationInfo quant_info = tensor.quantization_info(); - const int min_bound = quant_info.quantize(-1.f, RoundingPolicy::TO_NEAREST_UP); - const int max_bound = quant_info.quantize(1.f, RoundingPolicy::TO_NEAREST_UP); - std::uniform_int_distribution<> distribution(min_bound, max_bound); + std::pair<int, int> bounds = get_quantized_bounds(tensor.quantization_info(), -1.0f, 1.0f); + std::uniform_int_distribution<> distribution(bounds.first, bounds.second); library->fill(tensor, distribution, 0); } |