diff options
author | Matthew Bentham <matthew.bentham@arm.com> | 2019-04-26 16:57:29 +0100 |
---|---|---|
committer | Les Bell <les.bell@arm.com> | 2019-04-29 08:16:39 +0000 |
commit | c01b39146d4b67159ef95a2ac2f88f6536320890 (patch) | |
tree | d0b939cc870a20a1f2f36e636ba1db8a93279feb /src | |
parent | b85547024490d4c89538dcebfb0a2730e18a6d8d (diff) | |
download | armnn-c01b39146d4b67159ef95a2ac2f88f6536320890.tar.gz |
IVGCVSW-3007 ArmnnConverter: Fix bug with optional --input-tensor-shape argument
Change-Id: Icae21aa39bea57fd5de4278756cc620ffe684ed7
Signed-off-by: Matthew Bentham <matthew.bentham@arm.com>
Diffstat (limited to 'src')
-rw-r--r-- | src/armnnConverter/ArmnnConverter.cpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/armnnConverter/ArmnnConverter.cpp b/src/armnnConverter/ArmnnConverter.cpp index 67246fcd95..04cbb5dba2 100644 --- a/src/armnnConverter/ArmnnConverter.cpp +++ b/src/armnnConverter/ArmnnConverter.cpp @@ -196,8 +196,12 @@ int ParseCommandLineArgs(int argc, const char* argv[], return EXIT_FAILURE; } + if (!vm["input-tensor-shape"].empty()) + { + inputTensorShapeStrs = vm["input-tensor-shape"].as<std::vector<std::string>>(); + } + inputNames = vm["input-name"].as<std::vector<std::string>>(); - inputTensorShapeStrs = vm["input-tensor-shape"].as<std::vector<std::string>>(); outputNames = vm["output-name"].as<std::vector<std::string>>(); return EXIT_SUCCESS; |