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 /ethosu/vela/tflite_mapping.py | |
| 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>
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 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", |