diff options
Diffstat (limited to 'tests/InferenceTest.inl')
-rw-r--r-- | tests/InferenceTest.inl | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tests/InferenceTest.inl b/tests/InferenceTest.inl index 91a90f3820..b6087c5e5a 100644 --- a/tests/InferenceTest.inl +++ b/tests/InferenceTest.inl @@ -7,6 +7,8 @@ #include <armnn/Utils.hpp> #include <armnn/utility/Assert.hpp> #include <armnn/utility/NumericCast.hpp> +#include <armnnUtils/TContainer.hpp> + #include "CxxoptsUtils.hpp" #include <cxxopts/cxxopts.hpp> @@ -38,7 +40,7 @@ ClassifierTestCase<TTestCaseDatabase, TModel>::ClassifierTestCase( unsigned int label, std::vector<typename TModel::DataType> modelInput) : InferenceModelTestCase<TModel>( - model, testCaseId, std::vector<armnn::TContainer>{ modelInput }, { model.GetOutputSize() }) + model, testCaseId, std::vector<armnnUtils::TContainer>{ modelInput }, { model.GetOutputSize() }) , m_Label(label) , m_QuantizationParams(model.GetQuantizationParams()) , m_NumInferencesRef(numInferencesRef) |