diff options
author | Tim Hall <tim.hall@arm.com> | 2021-09-07 14:24:51 +0100 |
---|---|---|
committer | tim.hall <tim.hall@arm.com> | 2021-09-10 14:41:42 +0000 |
commit | b94bcd47e4137cc78f3df776e8fddec263ccbb63 (patch) | |
tree | ecf5f69f071fdefc5be6f27b16e3b811a6dcc799 /ethosu/vela/tflite_mapping.py | |
parent | 76b794e59b4ced500b0a0acb660e99caa78dbf90 (diff) | |
download | ethos-u-vela-b94bcd47e4137cc78f3df776e8fddec263ccbb63.tar.gz |
MLBEDSW-5174: Third-party custom ops not passed through correctly
- Re-added the CustomOptionsSerializer to the CUSTOM op TFLite mapping
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I51b141749ba223c132190077eed9e22fac798d2d
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 33117eb0..02e91c77 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -896,7 +896,7 @@ builtin_operator_map = { ), TFLITE_NO_INDICES, ), - BuiltinOperator.CUSTOM: (Op.Custom, None, TFLITE_NO_INDICES), + BuiltinOperator.CUSTOM: (Op.Custom, CustomOptionsSerializer(), TFLITE_NO_INDICES), } builtin_operator_inv_map = {v[0]: (k, v[1], v[2]) for k, v in builtin_operator_map.items()} |