diff options
| author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2021-04-16 14:36:22 +0200 |
|---|---|---|
| committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2021-04-19 08:14:26 +0000 |
| commit | 1634305e99082c285f2d3e14bdedf05b61f7be93 (patch) | |
| tree | fbe27a0e6cf46e2296507e01745cbeedeace8fd0 | |
| parent | a93f02629f9d69257f2b0010a7d6db12f965e889 (diff) | |
| download | ethos-u-vela-1634305e99082c285f2d3e14bdedf05b61f7be93.tar.gz | |
[MLBEDSW-4414] Fix verbose-operators for multiple custom ops
Fixed exception for --verbose-operators option when there are
multiple custom operators in the network.
Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: I5ab743d96a4e0367818fbe46cc47896c691d888c
| -rw-r--r-- | ethosu/vela/nn_graph.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py index 19734873..2d4b0c87 100644 --- a/ethosu/vela/nn_graph.py +++ b/ethosu/vela/nn_graph.py @@ -337,10 +337,13 @@ class Subgraph: attrs["kshape"] = [kshape[0], kshape[1]] attrs["type"] = op.type.name attrs.pop("use_cudnn_on_gpu", None) + custom_options = attrs.pop("custom_options", None) if attrs not in unique_ops: unique_ops.append(attrs) # print attributes in human readable format a = attrs.copy() + if custom_options is not None: + a["custom_options"] = custom_options s = a.pop("type") data_format = a.pop("data_format", None) if data_format and data_format != b"NHWC": |