diff options
Diffstat (limited to 'tests/ImageTensorGenerator/ImageTensorGenerator.hpp')
-rw-r--r-- | tests/ImageTensorGenerator/ImageTensorGenerator.hpp | 6 |
1 files changed, 4 insertions, 2 deletions
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 +} |