aboutsummaryrefslogtreecommitdiff
path: root/src/armnnCaffeParser
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnCaffeParser')
-rw-r--r--src/armnnCaffeParser/CaffeParser.cpp8
-rw-r--r--src/armnnCaffeParser/CaffeParser.hpp3
2 files changed, 11 insertions, 0 deletions
diff --git a/src/armnnCaffeParser/CaffeParser.cpp b/src/armnnCaffeParser/CaffeParser.cpp
index dfb9cec206..463f3eb2a5 100644
--- a/src/armnnCaffeParser/CaffeParser.cpp
+++ b/src/armnnCaffeParser/CaffeParser.cpp
@@ -5,6 +5,8 @@
#include "CaffeParser.hpp"
#include "RecordByRecordCaffeParser.hpp"
+#include "armnnCaffeParser/Version.hpp"
+
#include "armnn/Descriptors.hpp"
#include "armnn/INetwork.hpp"
#include "armnn/Utils.hpp"
@@ -35,6 +37,7 @@
#include <google/protobuf/wire_format.h>
#include <cmath>
+#include <iostream>
#include <sstream>
#include <queue>
#include <fcntl.h>
@@ -2250,6 +2253,11 @@ INetworkPtr ICaffeParser::CaffeParserImpl::CreateNetworkFromNetParameter(NetPara
return move(m_Network);
}
+const std::string ICaffeParser::CaffeParserImpl::GetVersion()
+{
+ return CAFFE_PARSER_VERSION;
+}
+
void ICaffeParser::CaffeParserImpl::Cleanup() {
// cleanup, in case we reuse this parser
m_InputShapes.clear();
diff --git a/src/armnnCaffeParser/CaffeParser.hpp b/src/armnnCaffeParser/CaffeParser.hpp
index f369d5f2f9..9f93569742 100644
--- a/src/armnnCaffeParser/CaffeParser.hpp
+++ b/src/armnnCaffeParser/CaffeParser.hpp
@@ -54,6 +54,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();
+
CaffeParserImpl();
virtual ~CaffeParserImpl() = default;