commit bceff2fb3fc68bb0aa88b886900c34b77340c826
author surmeh01 <surabhi.mehta@arm.com> Thu Mar 29 16:29:27 2018 +0100
committer surmeh01 <surabhi.mehta@arm.com> Thu Mar 29 16:29:27 2018 +0100
tree d867d3e090d58d3012dfbbac456e9ea8c7f789bc
parent 4fcda0101ec3d110c1d6d7bee5c83416b645528a

Release 18.03

tests/ExecuteNetwork/ExecuteNetwork.cpp

diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp
index 5e9e653..04ab195 100644
--- a/tests/ExecuteNetwork/ExecuteNetwork.cpp
+++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp
@@ -6,6 +6,9 @@
 #if defined(ARMNN_CAFFE_PARSER)
 #include "armnnCaffeParser/ICaffeParser.hpp"
 #endif
+#if defined(ARMNN_TF_PARSER)
+#include "armnnTfParser/ITfParser.hpp"
+#endif
 #include "Logging.hpp"
 #include "../InferenceTest.hpp"
 
@@ -232,8 +235,13 @@
     }
     else if (modelFormat.find("tensorflow") != std::string::npos)
     {
+#if defined(ARMNN_TF_PARSER)
+        return MainImpl<armnnTfParser::ITfParser, float>(modelPath.c_str(), isModelBinary, computeDevice,
+            inputName.c_str(), inputTensorShape.get(), inputTensorDataFilePath.c_str(), outputName.c_str());
+#else
         BOOST_LOG_TRIVIAL(fatal) << "Not built with Tensorflow parser support.";
         return 1;
+#endif
     }
     else
     {