diff options
Diffstat (limited to 'src/armnnTfParser')
-rwxr-xr-x | src/armnnTfParser/TfParser.cpp | 8 | ||||
-rw-r--r-- | src/armnnTfParser/TfParser.hpp | 3 |
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; |