diff options
| author | Ferran Balaguer <ferran.balaguer@arm.com> | 2018-12-28 18:15:24 +0000 |
|---|---|---|
| committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-12-31 14:56:00 +0000 |
| commit | fbdad03c927aa5d30deec6fa1a61eef10f8c265f (patch) | |
| tree | 0f435edc1178c470a6b9b7dcb3a408d486da5b3e /src/armnnTfParser/TfParser.hpp | |
| parent | d59180174bda6d533e233d733ec526d2db64a22e (diff) | |
| download | armnn-fbdad03c927aa5d30deec6fa1a61eef10f8c265f.tar.gz | |
IVGCVSW-2375 Add ParseAddN function to TfParser
* Unit tests in AddN.cpp
Change-Id: Ifb2fa1051d5d92c5d9a5ca751abee4e81ebe39c9
Diffstat (limited to 'src/armnnTfParser/TfParser.hpp')
| -rw-r--r-- | src/armnnTfParser/TfParser.hpp | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/src/armnnTfParser/TfParser.hpp b/src/armnnTfParser/TfParser.hpp index 55797471e2..0d1e497e29 100644 --- a/src/armnnTfParser/TfParser.hpp +++ b/src/armnnTfParser/TfParser.hpp @@ -129,6 +129,7 @@ private: bool HasParsedConstTensor(ParsedTfOperation* parsedTfOpPtr) const; ParsedTfOperationPtr ParseAdd(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); + ParsedTfOperationPtr ParseAddN(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseBiasAdd(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseConv2D(const tensorflow::NodeDef& nodeDef, const tensorflow::GraphDef& graphDef); ParsedTfOperationPtr ParseDepthwiseConv2D(const tensorflow::NodeDef& nodeDef,const tensorflow::GraphDef& graphDef); @@ -187,6 +188,31 @@ private: armnn::IConnectableLayer* const layer, const tensorflow::NodeDef& nodeDef); + armnn::IConnectableLayer* CreateAdditionLayer( + const tensorflow::NodeDef& nodeDef, + armnn::IOutputSlot* input0Slot, + armnn::IOutputSlot* input1Slot, + const std::string& layerName); + + armnn::IConnectableLayer* CreateAdditionLayer( + const tensorflow::NodeDef& nodeDef, + const OutputOfParsedTfOperation& opOne, + const OutputOfParsedTfOperation& opTwo, + unsigned int numberOfAddition); + + armnn::IConnectableLayer* CreateAdditionLayer( + const tensorflow::NodeDef& nodeDef, + armnn::IConnectableLayer* layerOne, + armnn::IConnectableLayer* layerTwo, + unsigned int numberOfAddition, + unsigned long numberOfLayersToConnect, + bool isOdd); + + armnn::IConnectableLayer* CreateAdditionLayer( + const tensorflow::NodeDef& nodeDef, + const OutputOfParsedTfOperation& op, + armnn::IConnectableLayer* layer); + static std::pair<armnn::LayerBindingId, armnn::TensorInfo> GetBindingInfo(const std::string& layerName, const char* bindingPointDesc, const std::unordered_map<std::string, BindingPointInfo>& nameToBindingInfo); |