diff options
Diffstat (limited to 'include/armnnCaffeParser/ICaffeParser.hpp')
-rw-r--r-- | include/armnnCaffeParser/ICaffeParser.hpp | 66 |
1 files changed, 0 insertions, 66 deletions
diff --git a/include/armnnCaffeParser/ICaffeParser.hpp b/include/armnnCaffeParser/ICaffeParser.hpp deleted file mode 100644 index 0e31ad4461..0000000000 --- a/include/armnnCaffeParser/ICaffeParser.hpp +++ /dev/null @@ -1,66 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// -#pragma once - -#include "armnn/Types.hpp" -#include "armnn/NetworkFwd.hpp" -#include "armnn/Tensor.hpp" -#include "armnn/INetwork.hpp" - -#include <memory> -#include <map> -#include <vector> - -namespace armnnCaffeParser -{ - -using BindingPointInfo = armnn::BindingPointInfo; - -class ICaffeParser; -using ICaffeParserPtr = std::unique_ptr<ICaffeParser, void(*)(ICaffeParser* parser)>; - -class ICaffeParser -{ -public: - static ICaffeParser* CreateRaw(); - static ICaffeParserPtr Create(); - static void Destroy(ICaffeParser* parser); - - /// Create the network from a protobuf text file on the disk. - armnn::INetworkPtr CreateNetworkFromTextFile( - const char* graphFile, - const std::map<std::string, armnn::TensorShape>& inputShapes, - const std::vector<std::string>& requestedOutputs); - - /// Create the network from a protobuf binary file on the disk. - armnn::INetworkPtr CreateNetworkFromBinaryFile( - const char* graphFile, - const std::map<std::string, armnn::TensorShape>& inputShapes, - const std::vector<std::string>& requestedOutputs); - - /// Create the network directly from protobuf text in a string. Useful for debugging/testin.g - armnn::INetworkPtr CreateNetworkFromString( - const char* protoText, - const std::map<std::string, armnn::TensorShape>& inputShapes, - const std::vector<std::string>& requestedOutputs); - - /// Retrieve binding info (layer id and tensor info) for the network input identified by the given layer name. - BindingPointInfo GetNetworkInputBindingInfo(const std::string& name) const; - - /// Retrieve binding info (layer id and tensor info) for the network output identified by the given layer name. - BindingPointInfo GetNetworkOutputBindingInfo(const std::string& name) const; - -private: - friend class CaffeParser; - friend class RecordByRecordCaffeParser; - - ICaffeParser(); - ~ICaffeParser(); - - class CaffeParserImpl; - std::unique_ptr<CaffeParserImpl> pCaffeParserImpl; -}; - -}
\ No newline at end of file |