aboutsummaryrefslogtreecommitdiff
path: root/tests/MultipleNetworksCifar10
diff options
context:
space:
mode:
authorJim Flynn <jim.flynn@arm.com>2019-05-03 12:54:26 +0100
committerRuomei Yan <ruomei.yan@arm.com>2019-05-07 09:21:19 +0000
commit2fd6100c7c034d6fc4420fbb5b615660803e46e9 (patch)
tree121f2faa4be9adeb2fedc4b9116426b5c9388421 /tests/MultipleNetworksCifar10
parent0eea37c05f82b299f6579ad119194f429147687e (diff)
downloadarmnn-2fd6100c7c034d6fc4420fbb5b615660803e46e9.tar.gz
IVGCVSW-3038 Move MakeInputTensors to armnnUtils
Change-Id: I4352a645badde788b9e33b3675a9cc9c0fc3dc33 Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to 'tests/MultipleNetworksCifar10')
-rw-r--r--tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp6
1 files changed, 2 insertions, 4 deletions
diff --git a/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp b/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp
index 006318f73a..fec78ac805 100644
--- a/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp
+++ b/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp
@@ -168,8 +168,6 @@ int main(int argc, char* argv[])
}
Cifar10Database cifar10(dataDir);
- using TContainer = boost::variant<std::vector<float>, std::vector<int>, std::vector<unsigned char>>;
-
for (unsigned int i = 0; i < 3; ++i)
{
// Loads test case data (including image data).
@@ -193,8 +191,8 @@ int main(int argc, char* argv[])
std::vector<TContainer> outputDataContainers = { outputs[k] };
status = runtime->EnqueueWorkload(networks[k].m_Network,
- MakeInputTensors(inputBindings, inputDataContainers),
- MakeOutputTensors(outputBindings, outputDataContainers));
+ armnnUtils::MakeInputTensors(inputBindings, inputDataContainers),
+ armnnUtils::MakeOutputTensors(outputBindings, outputDataContainers));
if (status == armnn::Status::Failure)
{
BOOST_LOG_TRIVIAL(fatal) << "armnn::IRuntime: Failed to enqueue workload";