diff options
Diffstat (limited to 'tests/ExecuteNetwork/ExecuteNetwork.cpp')
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index 5e9e6532cf..04ab195816 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 @@ int main(int argc, char* argv[]) } 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 { |