aboutsummaryrefslogtreecommitdiff
path: root/src/armnnOnnxParser
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnOnnxParser')
-rw-r--r--src/armnnOnnxParser/OnnxParser.cpp8
-rw-r--r--src/armnnOnnxParser/OnnxParser.hpp3
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*