diff options
| author | Saoirse Stewart <saoirse.stewart@arm.com> | 2019-02-27 11:07:57 +0000 |
|---|---|---|
| committer | Saoirse Stewart Arm <saoirse.stewart@arm.com> | 2019-02-27 13:16:23 +0000 |
| commit | 91c0eff2ff4ff384e013cb69cac1e07e28b9e2b1 (patch) | |
| tree | 9db521e088c92dd7b9d9e2e6dfd917559aab4745 /src/armnnTfParser/TfParser.hpp | |
| parent | dbfb8549d4aa80115a7049b3e94788fb7a474d9b (diff) | |
| download | armnn-91c0eff2ff4ff384e013cb69cac1e07e28b9e2b1.tar.gz | |
IVGCVSW-2598 Fix for constant axis issue for Tensorflow Parser
Change-Id: I8b081012529aed8e434273259c5a5ef7dc3afff7
Signed-off-by: Finn Williams <finn.williams@arm.com>
Signed-off-by: Saoirse Stewart <saoirse.stewart@arm.com>
Diffstat (limited to 'src/armnnTfParser/TfParser.hpp')
| -rw-r--r-- | src/armnnTfParser/TfParser.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnnTfParser/TfParser.hpp b/src/armnnTfParser/TfParser.hpp index 46da55f1d1..95ccf397c1 100644 --- a/src/armnnTfParser/TfParser.hpp +++ b/src/armnnTfParser/TfParser.hpp @@ -129,6 +129,8 @@ private: template<typename Type> bool HasParsedConstTensor(ParsedTfOperation* parsedTfOpPtr) const; + unsigned int GetConstInputIndex(const std::vector<OutputOfParsedTfOperation>& inputs); + 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); |