aboutsummaryrefslogtreecommitdiff
path: root/src/armnnCaffeParser/CaffeParser.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnCaffeParser/CaffeParser.cpp')
-rw-r--r--src/armnnCaffeParser/CaffeParser.cpp8
1 files changed, 8 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();