diff options
Diffstat (limited to 'src/armnnCaffeParser')
-rw-r--r-- | src/armnnCaffeParser/CaffeParser.cpp | 8 | ||||
-rw-r--r-- | src/armnnCaffeParser/CaffeParser.hpp | 3 |
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; |