From c01b39146d4b67159ef95a2ac2f88f6536320890 Mon Sep 17 00:00:00 2001
From: Matthew Bentham <matthew.bentham@arm.com>
Date: Fri, 26 Apr 2019 16:57:29 +0100
Subject: IVGCVSW-3007 ArmnnConverter: Fix bug with optional
 --input-tensor-shape argument

Change-Id: Icae21aa39bea57fd5de4278756cc620ffe684ed7
Signed-off-by: Matthew Bentham <matthew.bentham@arm.com>
---
 src/armnnConverter/ArmnnConverter.cpp | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

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;
-- 
cgit v1.2.1