diff options
author | telsoa01 <telmo.soares@arm.com> | 2018-08-31 09:22:23 +0100 |
---|---|---|
committer | telsoa01 <telmo.soares@arm.com> | 2018-08-31 09:22:23 +0100 |
commit | c577f2c6a3b4ddb6ba87a882723c53a248afbeba (patch) | |
tree | bd7d4c148df27f8be6649d313efb24f536b7cf34 /src/armnnCaffeParser/RecordByRecordCaffeParser.hpp | |
parent | 4c7098bfeab1ffe1cdc77f6c15548d3e73274746 (diff) | |
download | armnn-c577f2c6a3b4ddb6ba87a882723c53a248afbeba.tar.gz |
Release 18.08
Diffstat (limited to 'src/armnnCaffeParser/RecordByRecordCaffeParser.hpp')
-rw-r--r-- | src/armnnCaffeParser/RecordByRecordCaffeParser.hpp | 53 |
1 files changed, 53 insertions, 0 deletions
diff --git a/src/armnnCaffeParser/RecordByRecordCaffeParser.hpp b/src/armnnCaffeParser/RecordByRecordCaffeParser.hpp new file mode 100644 index 0000000000..f0855b4ecb --- /dev/null +++ b/src/armnnCaffeParser/RecordByRecordCaffeParser.hpp @@ -0,0 +1,53 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// See LICENSE file in the project root for full license information. +// + +#pragma once + +#include <string> +#include <vector> +#include <iostream> + +#include "caffe/proto/caffe.pb.h" + +#include "CaffeParser.hpp" + + + +namespace armnnCaffeParser +{ + +class NetParameterInfo; +class LayerParameterInfo; + + +class RecordByRecordCaffeParser : public CaffeParserBase +{ +public: + + /// Create the network from a protobuf binary file on disk + virtual armnn::INetworkPtr CreateNetworkFromBinaryFile( + const char* graphFile, + const std::map<std::string, armnn::TensorShape>& inputShapes, + const std::vector<std::string>& requestedOutputs) override; + + RecordByRecordCaffeParser(); + +private: + void ProcessLayers(const NetParameterInfo& netParameterInfo, + std::vector<LayerParameterInfo>& layerInfo, + const std::vector<std::string>& m_RequestedOutputs, + std::vector<const LayerParameterInfo*>& sortedNodes); + armnn::INetworkPtr LoadLayers(std::ifstream& ifs, + std::vector<const LayerParameterInfo *>& sortedNodes, + const NetParameterInfo& netParameterInfo); + std::vector<const LayerParameterInfo*> GetInputs( + const LayerParameterInfo& layerParam); + + std::map<std::string, const LayerParameterInfo*> m_CaffeLayersByTopName; + std::vector<std::string> m_RequestedOutputs; +}; + +} // namespace armnnCaffeParser + |