diff options
| author | surmeh01 <surabhi.mehta@arm.com> | 2018-03-29 16:29:27 +0100 |
|---|---|---|
| committer | surmeh01 <surabhi.mehta@arm.com> | 2018-03-29 16:29:27 +0100 |
| commit | bceff2fb3fc68bb0aa88b886900c34b77340c826 (patch) | |
| tree | d867d3e090d58d3012dfbbac456e9ea8c7f789bc /tests/ExecuteNetwork | |
| parent | 4fcda0101ec3d110c1d6d7bee5c83416b645528a (diff) | |
| download | armnn-bceff2fb3fc68bb0aa88b886900c34b77340c826.tar.gz | |
Release 18.03
Diffstat (limited to 'tests/ExecuteNetwork')
| -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 { |