diff options
Diffstat (limited to 'tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp')
-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"; |