From 264f7faa59709ffa8117541f5d55c99c5dba967d Mon Sep 17 00:00:00 2001
From: Jerry Ge <jerry.ge@arm.com>
Date: Fri, 21 Apr 2023 22:49:57 +0000
Subject: Add support for one dimension of size -1 in ReshapeOp

Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>

Change-Id: I0ef7607f4266296a1204c5cccdb5be36f345b5ba
---
 reference_model/src/subgraph_traverser.h | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'reference_model/src/subgraph_traverser.h')

diff --git a/reference_model/src/subgraph_traverser.h b/reference_model/src/subgraph_traverser.h
index 543b008..1cf582e 100644
--- a/reference_model/src/subgraph_traverser.h
+++ b/reference_model/src/subgraph_traverser.h
@@ -63,6 +63,10 @@ public:
     {
         return block->GetRegionName();
     }
+    TosaSerializationHandler* getTsh() const
+    {
+        return tsh;
+    }
     int getNumInputTensors() const;
     Tensor* getInputTensor(const unsigned int idx) const;
     Tensor* getInputTensorByName(const std::string name) const;
-- 
cgit v1.2.1