aboutsummaryrefslogtreecommitdiff
path: root/src/armnnTfParser
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnTfParser')
-rwxr-xr-xsrc/armnnTfParser/TfParser.cpp8
-rw-r--r--src/armnnTfParser/TfParser.hpp3
2 files changed, 11 insertions, 0 deletions
diff --git a/src/armnnTfParser/TfParser.cpp b/src/armnnTfParser/TfParser.cpp
index d13a277924..1e566fe943 100755
--- a/src/armnnTfParser/TfParser.cpp
+++ b/src/armnnTfParser/TfParser.cpp
@@ -5,6 +5,8 @@
#include "TfParser.hpp"
+#include "armnnTfParser/Version.hpp"
+
#include <armnn/TypesUtils.hpp>
#include <armnn/Descriptors.hpp>
@@ -25,6 +27,7 @@
#include <fmt/core.h>
#include <fmt/format.h>
+#include <iostream>
#include <numeric>
using namespace armnnUtils;
@@ -3734,4 +3737,9 @@ void ITfParser::TfParserImpl::TrackBindingPoint(IConnectableLayer* layer,
}
}
+const std::string ITfParser::TfParserImpl::GetVersion()
+{
+ return TF_PARSER_VERSION;
+}
+
} // namespace armnnTfParser
diff --git a/src/armnnTfParser/TfParser.hpp b/src/armnnTfParser/TfParser.hpp
index 5c04cceb82..31e074de93 100644
--- a/src/armnnTfParser/TfParser.hpp
+++ b/src/armnnTfParser/TfParser.hpp
@@ -88,6 +88,9 @@ public:
/// Retrieves binding info (layer id and tensor info) for the network output identified by the given layer name.
BindingPointInfo GetNetworkOutputBindingInfo(const std::string& name) const;
+ /// Retrieve version in X.Y.Z form
+ static const std::string GetVersion();
+
TfParserImpl();
~TfParserImpl() = default;