aboutsummaryrefslogtreecommitdiff
path: root/tests/InferenceModel.hpp
diff options
context:
space:
mode:
authorDavid Monahan <David.Monahan@arm.com>2021-10-22 12:57:28 +0100
committerDavid Monahan <david.monahan@arm.com>2021-10-22 13:53:18 +0000
commit6bb47a720ad33d0a330228b52e320a1022e4dfe2 (patch)
tree606434ca03951a79c828fb6371deb4e62fdcd858 /tests/InferenceModel.hpp
parent81ec994a3ebc8ad02c4a622846cf64b70e1182bd (diff)
downloadarmnn-6bb47a720ad33d0a330228b52e320a1022e4dfe2.tar.gz
IVGCVSW-6359 Create a single definition of TContainer
* Added a single definition of TContainer to include/armnn/Utils.hpp * Change all files which contained their own identical definitions of TContainer to use the new one Signed-off-by: David Monahan <David.Monahan@arm.com> Change-Id: I63e633693a430bbbd6a29001cafa19742ef8309a
Diffstat (limited to 'tests/InferenceModel.hpp')
-rw-r--r--tests/InferenceModel.hpp17
1 files changed, 9 insertions, 8 deletions
diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp
index cf3aae137e..e2cd5d9868 100644
--- a/tests/InferenceModel.hpp
+++ b/tests/InferenceModel.hpp
@@ -5,7 +5,9 @@
#pragma once
+
#include <armnn/ArmNN.hpp>
+#include <armnn/Utils.hpp>
#include <armnn/Threadpool.hpp>
#include <armnn/Logging.hpp>
#include <armnn/utility/Timer.hpp>
@@ -371,8 +373,7 @@ public:
using DataType = TDataType;
using Params = InferenceModelInternal::Params;
using QuantizationParams = InferenceModelInternal::QuantizationParams;
- using TContainer
- = mapbox::util::variant<std::vector<float>, std::vector<int>, std::vector<unsigned char>, std::vector<int8_t>>;
+
struct CommandLineOptions
{
@@ -583,8 +584,8 @@ public:
}
std::chrono::duration<double, std::milli> Run(
- const std::vector<TContainer>& inputContainers,
- std::vector<TContainer>& outputContainers)
+ const std::vector<armnn::TContainer>& inputContainers,
+ std::vector<armnn::TContainer>& outputContainers)
{
for (unsigned int i = 0; i < outputContainers.size(); ++i)
{
@@ -632,8 +633,8 @@ public:
std::tuple<unsigned int, std::chrono::duration<double, std::milli>> RunAsync(
armnn::experimental::IWorkingMemHandle& workingMemHandleRef,
- const std::vector<TContainer>& inputContainers,
- std::vector<TContainer>& outputContainers,
+ const std::vector<armnn::TContainer>& inputContainers,
+ std::vector<armnn::TContainer>& outputContainers,
unsigned int inferenceID)
{
for (unsigned int i = 0; i < outputContainers.size(); ++i)
@@ -683,8 +684,8 @@ public:
}
}
- void RunAsync(const std::vector<TContainer>& inputContainers,
- std::vector<TContainer>& outputContainers,
+ void RunAsync(const std::vector<armnn::TContainer>& inputContainers,
+ std::vector<armnn::TContainer>& outputContainers,
std::shared_ptr<armnn::IAsyncExecutionCallback> cb)
{
for (unsigned int i = 0; i < outputContainers.size(); ++i)