diff options
-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; |