TOSA: Add support for Identity operation
Added support for Identity operation.
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: If00b30528932f7531807ce3914d6c1875ab72fa4
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py
index b426792..1e733d5 100644
--- a/ethosu/vela/operation.py
+++ b/ethosu/vela/operation.py
@@ -189,7 +189,7 @@
GreaterEqual = OperatorInfo()
HardSwish = OperatorInfo(indices=NNG_IFM_INDICES)
HashtableLookup = OperatorInfo()
- Identity = OperatorInfo()
+ Identity = OperatorInfo(indices=NNG_IFM_INDICES)
If = OperatorInfo()
L2Norm = OperatorInfo()
L2Pool2D = OperatorInfo()