diff options
author | Jim Flynn <jim.flynn@arm.com> | 2019-05-03 12:54:26 +0100 |
---|---|---|
committer | Ruomei Yan <ruomei.yan@arm.com> | 2019-05-07 09:21:19 +0000 |
commit | 2fd6100c7c034d6fc4420fbb5b615660803e46e9 (patch) | |
tree | 121f2faa4be9adeb2fedc4b9116426b5c9388421 /tests/MultipleNetworksCifar10 | |
parent | 0eea37c05f82b299f6579ad119194f429147687e (diff) | |
download | armnn-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.cpp | 6 |
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"; |