aboutsummaryrefslogtreecommitdiff
path: root/tests/ImageTensorGenerator
diff options
context:
space:
mode:
authorJames Ward <james.ward@arm.com>2020-09-28 11:56:35 +0100
committerJames Ward <james.ward@arm.com>2020-09-30 17:13:58 +0000
commit6d9f5c57fe80b3b3c08294ddd52062e107151a15 (patch)
treea5aeca515a6f81545d82939ca4031df8677aed12 /tests/ImageTensorGenerator
parent156113cee2b869810092b42579c31c3b1fd910c5 (diff)
downloadarmnn-6d9f5c57fe80b3b3c08294ddd52062e107151a15.tar.gz
IVGCVSW-4519 Remove Boost Variant and apply_visitor variant
* replace boost::variant with mapbox::util::variant * replace boost::apply_visitor with mapbox::util::apply_visitor * replace boost::get with mapbox::util::get Signed-off-by: James Ward <james.ward@arm.com> Change-Id: I38460cabbcd5e56d4d61151bfe3dcb5681ce696e
Diffstat (limited to 'tests/ImageTensorGenerator')
-rw-r--r--tests/ImageTensorGenerator/ImageTensorGenerator.cpp11
1 files changed, 7 insertions, 4 deletions
diff --git a/tests/ImageTensorGenerator/ImageTensorGenerator.cpp b/tests/ImageTensorGenerator/ImageTensorGenerator.cpp
index 568ba1ee95..04419d5770 100644
--- a/tests/ImageTensorGenerator/ImageTensorGenerator.cpp
+++ b/tests/ImageTensorGenerator/ImageTensorGenerator.cpp
@@ -10,7 +10,7 @@
#include <Filesystem.hpp>
#include <boost/program_options.hpp>
-#include <boost/variant.hpp>
+#include <mapbox/variant.hpp>
#include <algorithm>
#include <fstream>
@@ -259,7 +259,7 @@ int main(int argc, char* argv[])
const unsigned int batchSize = 1;
const armnn::DataLayout outputLayout(cmdline.GetLayout());
- using TContainer = boost::variant<std::vector<float>, std::vector<int>, std::vector<uint8_t>>;
+ using TContainer = mapbox::util::variant<std::vector<float>, std::vector<int>, std::vector<uint8_t>>;
std::vector<TContainer> imageDataContainers;
const NormalizationParameters& normParams = GetNormalizationParameters(modelFormat, outputType);
try
@@ -291,8 +291,11 @@ int main(int argc, char* argv[])
imageTensorFile.open(outputPath, std::ofstream::out);
if (imageTensorFile.is_open())
{
- boost::apply_visitor([&imageTensorFile](auto&& imageData) { WriteImageTensorImpl(imageData, imageTensorFile); },
- imageDataContainers[0]);
+ mapbox::util::apply_visitor(
+ [&imageTensorFile](auto&& imageData){ WriteImageTensorImpl(imageData,imageTensorFile); },
+ imageDataContainers[0]
+ );
+
if (!imageTensorFile)
{
ARMNN_LOG(fatal) << "Failed to write to output file" << outputPath;