diff options
author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2022-08-19 16:06:04 +0200 |
---|---|---|
committer | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2022-08-31 15:32:24 +0200 |
commit | 8ddd4899892dace88306b3b155dbf47cc47fa4cd (patch) | |
tree | 1322377438b283ac8862696f8232ef48cd98a0ea /ethosu/vela/operation.py | |
parent | 1bbd06bf538ff9d290e9703b6bac05b9617f2ab0 (diff) | |
download | ethos-u-vela-8ddd4899892dace88306b3b155dbf47cc47fa4cd.tar.gz |
MLBEDSW-6832 PReLU support in Vela
Added PReLU support in graph optimiser.
Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: I3a188675e3edcdf0b4a4bfcdd134fda0bf8a560f
Diffstat (limited to 'ethosu/vela/operation.py')
-rw-r--r-- | ethosu/vela/operation.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 47f4fe0f..54e823a8 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -229,7 +229,7 @@ class Op(Enum): 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) |