diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/ImagePreprocessor.cpp | 5 | ||||
-rw-r--r-- | tests/ImageTensorGenerator/ImageTensorGenerator.hpp | 6 |
2 files changed, 8 insertions, 3 deletions
diff --git a/tests/ImagePreprocessor.cpp b/tests/ImagePreprocessor.cpp index 74bc943ee8..f0184e466e 100644 --- a/tests/ImagePreprocessor.cpp +++ b/tests/ImagePreprocessor.cpp @@ -2,11 +2,14 @@ // Copyright © 2017 Arm Ltd. All rights reserved. // SPDX-License-Identifier: MIT // + #include "InferenceTestImage.hpp" #include "ImagePreprocessor.hpp" -#include "Permute.hpp" + #include <armnn/TypesUtils.hpp> +#include <armnnUtils/Permute.hpp> + #include <boost/numeric/conversion/cast.hpp> #include <boost/assert.hpp> #include <boost/format.hpp> diff --git a/tests/ImageTensorGenerator/ImageTensorGenerator.hpp b/tests/ImageTensorGenerator/ImageTensorGenerator.hpp index 3e164bc9e0..b1cb5e36f5 100644 --- a/tests/ImageTensorGenerator/ImageTensorGenerator.hpp +++ b/tests/ImageTensorGenerator/ImageTensorGenerator.hpp @@ -4,9 +4,11 @@ // #include "../InferenceTestImage.hpp" -#include "Permute.hpp" + #include <armnn/TypesUtils.hpp> +#include <armnnUtils/Permute.hpp> + #include <algorithm> #include <fstream> #include <iterator> @@ -178,4 +180,4 @@ template <> void WriteImageTensorImpl<uint8_t>(const std::vector<uint8_t>& imageData, std::ofstream& imageTensorFile) { std::copy(imageData.begin(), imageData.end(), std::ostream_iterator<int>(imageTensorFile, " ")); -}
\ No newline at end of file +} |