aboutsummaryrefslogtreecommitdiff
path: root/python/pyarmnn/src
diff options
context:
space:
mode:
authorNikhil Raj <nikhil.raj@arm.com>2021-04-19 16:59:48 +0100
committerNikhil Raj <nikhil.raj@arm.com>2021-04-27 17:37:11 +0100
commit5d955cf70ae0c5558d4f431f0fc6bd4552cd43a5 (patch)
tree4fb59200899808b8b008d6f48322d0d799b8b631 /python/pyarmnn/src
parent4a621c43174b6bdd9dc0bff839b245bc2139d6a6 (diff)
downloadarmnn-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 'python/pyarmnn/src')
-rw-r--r--python/pyarmnn/src/pyarmnn/__init__.py13
-rw-r--r--python/pyarmnn/src/pyarmnn/swig/armnn_tfparser.i102
2 files changed, 0 insertions, 115 deletions
diff --git a/python/pyarmnn/src/pyarmnn/__init__.py b/python/pyarmnn/src/pyarmnn/__init__.py
index 410e66be11..5cb8bfb6cd 100644
--- a/python/pyarmnn/src/pyarmnn/__init__.py
+++ b/python/pyarmnn/src/pyarmnn/__init__.py
@@ -22,19 +22,6 @@ except ImportError as err:
raise RuntimeError(message)
try:
- from ._generated.pyarmnn_tfparser import ITfParser
-except ImportError as err:
- logger = logging.getLogger(__name__)
- message = "Your ArmNN library instance does not support TF models parser functionality. "
- logger.warning("%s Skipped ITfParser import.", message)
- logger.debug(str(err))
-
-
- def ITfParser():
- """In case people try importing without having Arm NN built with this parser."""
- raise RuntimeError(message)
-
-try:
from ._generated.pyarmnn_tfliteparser import ITfLiteParser, TfLiteParserOptions
except ImportError as err:
logger = logging.getLogger(__name__)
diff --git a/python/pyarmnn/src/pyarmnn/swig/armnn_tfparser.i b/python/pyarmnn/src/pyarmnn/swig/armnn_tfparser.i
deleted file mode 100644
index 03729abbf8..0000000000
--- a/python/pyarmnn/src/pyarmnn/swig/armnn_tfparser.i
+++ /dev/null
@@ -1,102 +0,0 @@
-//
-// Copyright © 2020 Arm Ltd. All rights reserved.
-// SPDX-License-Identifier: MIT
-//
-%module pyarmnn_tfparser
-%{
-#define SWIG_FILE_WITH_INIT
-#include "armnnTfParser/ITfParser.hpp"
-#include "armnn/INetwork.hpp"
-%}
-
-//typemap definitions and other common stuff
-%include "standard_header.i"
-
-namespace std {
- %template(BindingPointInfo) pair<int, armnn::TensorInfo>;
- %template(MapStringTensorShape) map<std::string, armnn::TensorShape>;
- %template(StringVector) vector<string>;
-}
-
-namespace armnnTfParser
-{
-%feature("docstring",
-"
-Interface for creating a parser object using TensorFlow (https://www.tensorflow.org/) frozen pb files.
-
-Parsers are used to automatically construct Arm NN graphs from model files.
-
-") ITfParser;
-%nodefaultctor ITfParser;
-class ITfParser
-{
-public:
- %feature("docstring",
- "
- Retrieve binding info (layer id and `TensorInfo`) for the network input identified by the given layer name.
-
- Args:
- name (str): Name of the input.
-
- Returns:
- tuple: (`int`, `TensorInfo`).
- ") GetNetworkInputBindingInfo;
- std::pair<int, armnn::TensorInfo> GetNetworkInputBindingInfo(const std::string& name);
-
- %feature("docstring",
- "
- Retrieve binding info (layer id and `TensorInfo`) for the network output identified by the given layer name.
-
- Args:
- name (str): Name of the output.
-
- Returns:
- tuple: (`int`, `TensorInfo`).
- ") GetNetworkOutputBindingInfo;
- std::pair<int, armnn::TensorInfo> GetNetworkOutputBindingInfo(const std::string& name);
-};
-
-%extend ITfParser {
- // This is not a substitution of the default constructor of the Armnn class. It tells swig to create custom __init__
- // method for ITfParser python object that will use static factory method to do the job.
-
- ITfParser() {
- return armnnTfParser::ITfParser::CreateRaw();
- }
-
- // The following does not replace a real destructor of the Armnn class.
- // It creates a functions that will be called when swig object goes out of the scope to clean resources.
- // so the user doesn't need to call ITfParser::Destroy himself.
- // $self` is a pointer to extracted ArmNN ITfParser object.
-
- ~ITfParser() {
- armnnTfParser::ITfParser::Destroy($self);
- }
-
- %feature("docstring",
- "
- Create the network from a pb Protocol buffer file.
-
- Args:
- graphFile (str): Path to the tf model to be parsed.
- inputShapes (dict): A dict containing the input name as a key and `TensorShape` as a value.
- requestedOutputs (list of str): A list of the output tensor names.
-
- Returns:
- INetwork: Parsed network.
-
- Raises:
- RuntimeError: If model file was not found.
- ") CreateNetworkFromBinaryFile;
- %newobject CreateNetworkFromBinaryFile;
- armnn::INetwork* CreateNetworkFromBinaryFile(const char* graphFile,
- const std::map<std::string, armnn::TensorShape>& inputShapes,
- const std::vector<std::string>& requestedOutputs) {
- return $self->CreateNetworkFromBinaryFile(graphFile, inputShapes, requestedOutputs).release();
- }
-
-}
-
-}
-// Clear exception typemap.
-%exception;