diff options
Diffstat (limited to 'src/armnn/test/TensorHelpers.hpp')
-rw-r--r-- | src/armnn/test/TensorHelpers.hpp | 28 |
1 files changed, 14 insertions, 14 deletions
diff --git a/src/armnn/test/TensorHelpers.hpp b/src/armnn/test/TensorHelpers.hpp index 35e471e0f2..3f8589353c 100644 --- a/src/armnn/test/TensorHelpers.hpp +++ b/src/armnn/test/TensorHelpers.hpp @@ -4,23 +4,21 @@ // #pragma once -#include <armnn/TensorFwd.hpp> -#include <boost/test/unit_test.hpp> -#include <boost/multi_array.hpp> -#include <vector> -#include <array> +#include <armnn/Tensor.hpp> + +#include <QuantizeHelper.hpp> #include <boost/assert.hpp> -#include <boost/test/tools/floating_point_comparison.hpp> +#include <boost/multi_array.hpp> +#include <boost/numeric/conversion/cast.hpp> #include <boost/random/uniform_real_distribution.hpp> #include <boost/random/mersenne_twister.hpp> -#include <boost/numeric/conversion/cast.hpp> - -#include <armnn/Tensor.hpp> - -#include <backendsCommon/test/QuantizeHelper.hpp> +#include <boost/test/tools/floating_point_comparison.hpp> +#include <boost/test/unit_test.hpp> +#include <array> #include <cmath> +#include <vector> constexpr float g_FloatCloseToZeroTolerance = 1.0e-6f; @@ -235,7 +233,9 @@ boost::multi_array<T, n> MakeRandomTensor(const armnn::TensorInfo& tensorInfo, { init[i] = dist(gen); } - float qScale = tensorInfo.GetQuantizationScale(); - int32_t qOffset = tensorInfo.GetQuantizationOffset(); - return MakeTensor<T, n>(tensorInfo, QuantizedVector<T>(qScale, qOffset, init)); + + const float qScale = tensorInfo.GetQuantizationScale(); + const int32_t qOffset = tensorInfo.GetQuantizationOffset(); + + return MakeTensor<T, n>(tensorInfo, armnnUtils::QuantizedVector<T>(init, qScale, qOffset)); } |