diff options
| -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()} |