From b94bcd47e4137cc78f3df776e8fddec263ccbb63 Mon Sep 17 00:00:00 2001 From: Tim Hall Date: Tue, 7 Sep 2021 14:24:51 +0100 Subject: 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 Change-Id: I51b141749ba223c132190077eed9e22fac798d2d --- ethosu/vela/tflite_mapping.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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()} -- cgit v1.2.1