aboutsummaryrefslogtreecommitdiff
path: root/python/pyarmnn/src/pyarmnn/swig/armnn_caffeparser.i
diff options
context:
space:
mode:
Diffstat (limited to 'python/pyarmnn/src/pyarmnn/swig/armnn_caffeparser.i')
-rw-r--r--python/pyarmnn/src/pyarmnn/swig/armnn_caffeparser.i103
1 files changed, 0 insertions, 103 deletions
diff --git a/python/pyarmnn/src/pyarmnn/swig/armnn_caffeparser.i b/python/pyarmnn/src/pyarmnn/swig/armnn_caffeparser.i
deleted file mode 100644
index 538b486aad..0000000000
--- a/python/pyarmnn/src/pyarmnn/swig/armnn_caffeparser.i
+++ /dev/null
@@ -1,103 +0,0 @@
-//
-// Copyright © 2020 Arm Ltd. All rights reserved.
-// SPDX-License-Identifier: MIT
-//
-%module pyarmnn_caffeparser
-%{
-#define SWIG_FILE_WITH_INIT
-#include "armnnCaffeParser/ICaffeParser.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 armnnCaffeParser
-{
-
-%feature("docstring",
-"
-Interface for creating a parser object using Caffe (http://caffe.berkeleyvision.org/) caffemodel files.
-
-Parsers are used to automatically construct Arm NN graphs from model files.
-
-") ICaffeParser;
-
-%nodefaultctor ICaffeParser;
-class ICaffeParser
-{
-public:
- // Documentation
- %feature("docstring",
- "
- Retrieve binding info (layer id and tensor info) for the network input identified by the given layer name.
-
- Args:
- name (str): Name of the input.
-
- Returns:
- tuple: (`int`, `TensorInfo`)
- ") GetNetworkInputBindingInfo;
-
- %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> GetNetworkInputBindingInfo(const std::string& name);
- std::pair<int, armnn::TensorInfo> GetNetworkOutputBindingInfo(const std::string& name);
-};
-
-%extend ICaffeParser {
- // This is not a substitution of the default constructor of the Armnn class. It tells swig to create custom __init__
- // method for ICaffeParser python object that will use static factory method to do the job.
-
- ICaffeParser() {
- return armnnCaffeParser::ICaffeParser::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 ICaffeParser::Destroy himself.
- // $self` is a pointer to extracted ArmNN ICaffeParser object.
-
- ~ICaffeParser() {
- armnnCaffeParser::ICaffeParser::Destroy($self);
- }
-
- %feature("docstring",
- "
- Create the network from a Caffe caffemodel binary file on disk.
-
- Args:
- graphFile: Path to the caffe model to be parsed.
- inputShapes (tuple): (`string`, `TensorShape`) A tuple containing the input name and TensorShape information for the network.
- requestedOutputs (list): A list of the output tensor names.
-
- Returns:
- INetwork: INetwork object for the parsed Caffe model.
- ") 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;