From 91c0eff2ff4ff384e013cb69cac1e07e28b9e2b1 Mon Sep 17 00:00:00 2001
From: Saoirse Stewart <saoirse.stewart@arm.com>
Date: Wed, 27 Feb 2019 11:07:57 +0000
Subject: 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>
---
 src/armnnTfParser/TfParser.hpp | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'src/armnnTfParser/TfParser.hpp')

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);
-- 
cgit v1.2.1