diff options
Diffstat (limited to 'src/armnnOnnxParser')
-rw-r--r-- | src/armnnOnnxParser/OnnxParser.cpp | 8 | ||||
-rw-r--r-- | src/armnnOnnxParser/OnnxParser.hpp | 3 |
2 files changed, 11 insertions, 0 deletions
diff --git a/src/armnnOnnxParser/OnnxParser.cpp b/src/armnnOnnxParser/OnnxParser.cpp index b4e7133239..81d9e3d240 100644 --- a/src/armnnOnnxParser/OnnxParser.cpp +++ b/src/armnnOnnxParser/OnnxParser.cpp @@ -4,6 +4,8 @@ // #include "OnnxParser.hpp" +#include "armnnOnnxParser/Version.hpp" + #include <armnn/Descriptors.hpp> #include <armnn/utility/Assert.hpp> #include <armnn/utility/NumericCast.hpp> @@ -14,6 +16,7 @@ #include <google/protobuf/text_format.h> #include <google/protobuf/io/zero_copy_stream_impl.h> +#include <iostream> #include <numeric> using namespace armnn; @@ -1849,4 +1852,9 @@ std::vector<std::string> OnnxParserImpl::GetOutputs(ModelPtr& model) return outputNames; } +const std::string OnnxParserImpl::GetVersion() +{ + return ONNX_PARSER_VERSION; +} + } // namespace armnnOnnxParser diff --git a/src/armnnOnnxParser/OnnxParser.hpp b/src/armnnOnnxParser/OnnxParser.hpp index 0db93248bc..7716e50fff 100644 --- a/src/armnnOnnxParser/OnnxParser.hpp +++ b/src/armnnOnnxParser/OnnxParser.hpp @@ -61,6 +61,9 @@ public: /// Retrieve outputs names static std::vector<std::string> GetOutputs(ModelPtr& model); + /// Retrieve version in X.Y.Z form + static const std::string GetVersion(); + private: /// Parses a ModelProto loaded into memory from one of the other CreateNetwork* |