commit b94bcd47e4137cc78f3df776e8fddec263ccbb63
author Tim Hall <tim.hall@arm.com> Tue Sep 07 14:24:51 2021 +0100
committer tim.hall <tim.hall@arm.com> Fri Sep 10 14:41:42 2021 +0000
tree ecf5f69f071fdefc5be6f27b16e3b811a6dcc799
parent 76b794e59b4ced500b0a0acb660e99caa78dbf90

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

ethosu/vela/tflite_mapping.py

diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py
index 33117eb..02e91c7 100644
--- a/ethosu/vela/tflite_mapping.py
+++ b/ethosu/vela/tflite_mapping.py
@@ -896,7 +896,7 @@
         ),
         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()}