diff options
author | Tim Hall <tim.hall@arm.com> | 2020-04-28 20:13:40 +0100 |
---|---|---|
committer | Tim Hall <tim.hall@arm.com> | 2020-06-18 17:53:52 +0100 |
commit | 10349e70e994130d9e1ac8b72f32575025b2b453 (patch) | |
tree | 64d4c1f0010be643d6b62bc367f86b2565924d47 | |
parent | 65da3795cb4f7e682fc849c7a384e6b9795863f1 (diff) | |
download | ethos-u-vela-10349e70e994130d9e1ac8b72f32575025b2b453.tar.gz |
MLBEDSW-2030: PReLU is not mapped for passthrough
- Added missing prelu op to tflite mapping
Change-Id: I1cb846fc044f0a5a78651569383a552d1cccbb52
Signed-off-by: Tim Hall <tim.hall@arm.com>
-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 c6981aa7..e2b90765 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -539,7 +539,7 @@ builtin_operator_map = { ), ), ), - # BuiltinOperator.PRELU : "", + BuiltinOperator.PRELU : ("Prelu", None), BuiltinOperator.MAXIMUM: ("Maximum", OptionsSerializer("MaximumMinimumOptions")), BuiltinOperator.ARG_MAX: ( "ArgMax", |