commit 8ddd4899892dace88306b3b155dbf47cc47fa4cd
author Fredrik Svedberg <fredrik.svedberg@arm.com> Fri Aug 19 16:06:04 2022 +0200
committer Fredrik Svedberg <fredrik.svedberg@arm.com> Wed Aug 31 15:32:24 2022 +0200
tree 1322377438b283ac8862696f8232ef48cd98a0ea
parent 1bbd06bf538ff9d290e9703b6bac05b9617f2ab0

MLBEDSW-6832 PReLU support in Vela

Added PReLU support in graph optimiser.

Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: I3a188675e3edcdf0b4a4bfcdd134fda0bf8a560f

ethosu/vela/operation.py

diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py
index 47f4fe0..54e823a 100644
--- a/ethosu/vela/operation.py
+++ b/ethosu/vela/operation.py
@@ -229,7 +229,7 @@
     PadV2 = OperatorInfo()
     Placeholder = OperatorInfo()  # Only used in CPU subgraphs
     Pow = OperatorInfo()
-    Prelu = OperatorInfo()
+    Prelu = OperatorInfo(indices=NNG_IFM_IFM2_INDICES)
     Prod = OperatorInfo()
     Quantize = OperatorInfo(indices=NNG_IFM_INDICES)
     QuantizedAvgPool = OperatorInfo(block_type=NpuBlockType.Pooling, indices=NNG_IFM_INDICES)