diff options
| author | Dwight Lidman <dwight.lidman@arm.com> | 2020-10-02 14:55:45 +0200 |
|---|---|---|
| committer | Dwight Lidman <dwight.lidman@arm.com> | 2020-10-16 11:33:44 +0200 |
| commit | c6ac1944d9934faf6d22825cdd3273afe55432a4 (patch) | |
| tree | a49ffe4db71486271737bd88897a38fe0d7f9e6a /ethosu/vela/operation.py | |
| parent | 5f2ea2ff77b62d12fc46ae8d08c7815a2a9caae2 (diff) | |
| download | ethos-u-vela-c6ac1944d9934faf6d22825cdd3273afe55432a4.tar.gz | |
MLBEDSW-3004: UnpackReshaped can't be serialised
This commit fixes a bug where a rewritten Unpack
operator is placed on the CPU and crashes Vela
during serialisation due to the type having
changed and there not being a mapping for the
modified op type.
The solution is to move the fixup_unpack_output
function to the graph optimisation pass B,
allowing the supported op check to run before it.
Signed-off-by: Dwight Lidman <dwight.lidman@arm.com>
Change-Id: Ic6bd4c70a478fd61adf377cb487f5b9253130314
Diffstat (limited to 'ethosu/vela/operation.py')
| -rw-r--r-- | ethosu/vela/operation.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 710511c6..6e5b4820 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -314,7 +314,7 @@ def get_slice_offsets(input_shape, offset_tens, offset_mask, is_begin=True): class Operation: """Class representing a Neural Network operation. Has a name, a type, -input and output tensors, as well as an attribute dictionary.""" + input and output tensors, as well as an attribute dictionary.""" __slots__ = ( "type", |