diff options
Diffstat (limited to 'tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp')
| -rw-r--r-- | tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp b/tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp index d1d31f48a1..c08d88e053 100644 --- a/tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp +++ b/tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp @@ -9,7 +9,7 @@ #include "armnnDeserializer/IDeserializer.hpp" #include <armnnUtils/Filesystem.hpp> -#include <armnn/Utils.hpp> +#include <armnnUtils/TContainer.hpp> #include <cxxopts/cxxopts.hpp> #include <map> @@ -325,8 +325,8 @@ int main(int argc, char* argv[]) const std::string imageName = imageEntry.first; std::cout << "Processing image: " << imageName << "\n"; - vector<armnn::TContainer> inputDataContainers; - vector<armnn::TContainer> outputDataContainers; + vector<armnnUtils::TContainer> inputDataContainers; + vector<armnnUtils::TContainer> outputDataContainers; auto imagePath = pathToDataDir / fs::path(imageName); switch (inputTensorDataType) @@ -370,7 +370,7 @@ int main(int argc, char* argv[]) ARMNN_LOG(fatal) << "armnn::IRuntime: Failed to enqueue workload for image: " << imageName; } - checker.AddImageResult<armnn::TContainer>(imageName, outputDataContainers); + checker.AddImageResult<armnnUtils::TContainer>(imageName, outputDataContainers); } } else |