diff options
Diffstat (limited to 'tests/NetworkExecutionUtils')
| -rw-r--r-- | tests/NetworkExecutionUtils/NetworkExecutionUtils.cpp | 2 | ||||
| -rw-r--r-- | tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp | 4 |
2 files changed, 3 insertions, 3 deletions
diff --git a/tests/NetworkExecutionUtils/NetworkExecutionUtils.cpp b/tests/NetworkExecutionUtils/NetworkExecutionUtils.cpp index 323e9fb812..6c74aaa6ed 100644 --- a/tests/NetworkExecutionUtils/NetworkExecutionUtils.cpp +++ b/tests/NetworkExecutionUtils/NetworkExecutionUtils.cpp @@ -228,7 +228,7 @@ void TensorPrinter::WriteToFile(const std::vector<T>& values) } } -void PopulateTensorWithData(armnn::TContainer& tensorData, +void PopulateTensorWithData(armnnUtils::TContainer& tensorData, unsigned int numElements, const std::string& dataTypeStr, const armnn::Optional<QuantizationParams>& qParams, diff --git a/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp b/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp index d9e2459703..bc2868ab35 100644 --- a/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp +++ b/tests/NetworkExecutionUtils/NetworkExecutionUtils.hpp @@ -9,7 +9,7 @@ #include <armnn/Types.hpp> #include <armnn/Logging.hpp> #include <armnn/utility/StringUtils.hpp> -#include <armnn/Utils.hpp> +#include <armnnUtils/TContainer.hpp> #include <iostream> #include <fstream> @@ -53,7 +53,7 @@ private: using QuantizationParams = std::pair<float, int32_t>; -void PopulateTensorWithData(armnn::TContainer& tensorData, +void PopulateTensorWithData(armnnUtils::TContainer& tensorData, unsigned int numElements, const std::string& dataTypeStr, const armnn::Optional<QuantizationParams>& qParams, |