commit 4965faee41300393cd8d74da4b399fa4c4ee9030
author Ayaan Masood <Ayaan.Masood@arm.com> Wed Jun 29 11:30:57 2022 +0100
committer Ayaan Masood <Ayaan.Masood@arm.com> Wed Jun 29 11:30:57 2022 +0100
tree 1054d6f89be70ec471007132dec97d325ecc0067
parent 68b8f2f9457d56df3211be5318e3682332bcefbf

MLBEDSW-6313 Static optimisation for Shape OP

*Shape OP value is available at compile time hence
it can be optimised
*Disconnected shape OP at compile time from parent
tensor
*Transformed shape OP tensor into constant

Change-Id: I0a024269e2b592c6146dd72e62d7a41951fb727a
Signed-off-by: Ayaan Masood <Ayaan.Masood@arm.com>

ethosu/vela/operation.py

diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py
index 8e06b1e..6b6671b 100644
--- a/ethosu/vela/operation.py
+++ b/ethosu/vela/operation.py
@@ -261,7 +261,7 @@
     SegmentSum = OperatorInfo()
     Select = OperatorInfo()
     SelectV2 = OperatorInfo()
-    Shape = OperatorInfo()
+    Shape = OperatorInfo(indices=NNG_IFM_INDICES)
     Sigmoid = OperatorInfo(indices=NNG_IFM_INDICES)
     SignBit = OperatorInfo()
     Sin = OperatorInfo()