diff options
author | Nikhil Raj <nikhil.raj@arm.com> | 2021-04-19 16:59:48 +0100 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2021-04-27 17:37:11 +0100 |
commit | 5d955cf70ae0c5558d4f431f0fc6bd4552cd43a5 (patch) | |
tree | 4fb59200899808b8b008d6f48322d0d799b8b631 /include/armnnTfParser/ITfParser.hpp | |
parent | 4a621c43174b6bdd9dc0bff839b245bc2139d6a6 (diff) | |
download | armnn-5d955cf70ae0c5558d4f431f0fc6bd4552cd43a5.tar.gz |
IVGCVSW-5721 Remove the Tensorflow Parser from ArmNN
Signed-off-by: Nikhil Raj <nikhil.raj@arm.com>
Change-Id: Ida37d3ee3a1af0c75aa905199bd861726c646846
Diffstat (limited to 'include/armnnTfParser/ITfParser.hpp')
-rw-r--r-- | include/armnnTfParser/ITfParser.hpp | 77 |
1 files changed, 0 insertions, 77 deletions
diff --git a/include/armnnTfParser/ITfParser.hpp b/include/armnnTfParser/ITfParser.hpp deleted file mode 100644 index 91e4cb39bf..0000000000 --- a/include/armnnTfParser/ITfParser.hpp +++ /dev/null @@ -1,77 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// -#pragma once - -#include "armnn/Types.hpp" -#include "armnn/Tensor.hpp" -#include "armnn/INetwork.hpp" - -#include <map> -#include <memory> -#include <unordered_map> -#include <vector> - -namespace armnnTfParser -{ - -using BindingPointInfo = armnn::BindingPointInfo; - -class ITfParser; -using ITfParserPtr = std::unique_ptr<ITfParser, void(*)(ITfParser* parser)>; - -/// Parses a directed acyclic graph from a tensorflow protobuf file. -class ITfParser -{ -public: - static ITfParser* CreateRaw(); - static ITfParserPtr Create(); - static void Destroy(ITfParser* 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/testing. - 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: - template <typename T> - friend class ParsedConstTfOperation; - friend class ParsedMatMulTfOperation; - friend class ParsedMulTfOperation; - friend class ParsedTfOperation; - friend class SingleLayerParsedTfOperation; - friend class DeferredSingleLayerParsedTfOperation; - friend class ParsedIdentityTfOperation; - - template <template<typename> class OperatorType, typename T> - friend struct MakeTfOperation; - - - ITfParser(); - ~ITfParser(); - - struct TfParserImpl; - std::unique_ptr<TfParserImpl> pTfParserImpl; -}; - -} |