From c8310b1432f7a77df3c95e8ecf8248c8a953b411 Mon Sep 17 00:00:00 2001 From: Tim Hall Date: Wed, 17 Jun 2020 14:53:11 +0100 Subject: MLBEDSW-2528: MLCE-219: Custom operator pass through - Fixed custom operator pass through - Added error printing functions for operators and tensor - Minor cleanup of custom exception handling Signed-off-by: Tim Hall Change-Id: Idf295df1e4c544381dc480244d880c32fb285e38 --- ethosu/vela/npu_serialisation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ethosu/vela/npu_serialisation.py') diff --git a/ethosu/vela/npu_serialisation.py b/ethosu/vela/npu_serialisation.py index 08dc0d38..18d38f3f 100644 --- a/ethosu/vela/npu_serialisation.py +++ b/ethosu/vela/npu_serialisation.py @@ -141,7 +141,7 @@ def rewrite_npu_call_ops(nng, sg, arch): for op in ps.ops: if op.type == "NpuOp": callee = op.attrs["subgraph"] - op.attrs["custom_options"] = {"type": op.type} + op.attrs["custom_type"] = op.type sz = 0 for tens in [callee.scratch_tensor, callee.flash_tensor, callee.command_stream_tensor]: -- cgit v1.2.1