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/tflite_mapping.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/tflite_mapping.py')
-rw-r--r-- | ethosu/vela/tflite_mapping.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py index c515d23a..3ccedc73 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -732,7 +732,7 @@ builtin_operator_map = { ), TFLITE_NO_INDICES, ), - BuiltinOperator.PRELU: (Op.Prelu, None, TFLITE_NO_INDICES), + BuiltinOperator.PRELU: (Op.Prelu, None, TFLITE_IFM_IFM2_INDICES), BuiltinOperator.MAXIMUM: (Op.Maximum, OptionsSerializer("MaximumMinimumOptions"), TFLITE_IFM_IFM2_INDICES), BuiltinOperator.ARG_MAX: ( Op.ArgMax, |