diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-02-04 16:28:29 +0100 |
---|---|---|
committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-02-08 14:24:41 +0100 |
commit | da2b0030220e87788573a724979626aa92afd13e (patch) | |
tree | 66b0400f90e6854129005e08232f7db94c64fd14 /ethosu/vela/operation.py | |
parent | 455e20e5ed0d5ce141a921e67f0219e55044e6e1 (diff) | |
download | ethos-u-vela-da2b0030220e87788573a724979626aa92afd13e.tar.gz |
MLBEDSW-3937 Fix check for NHCWB16 for FC
Fix check for NHCWB16 for modifying FC input.
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: Ie50c32ca079afadd0af9b7b909820794ceee373c
Diffstat (limited to 'ethosu/vela/operation.py')
-rw-r--r-- | ethosu/vela/operation.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 09371b7a..b297bed0 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -739,7 +739,7 @@ class Operation: if self.type == Op.Softmax: self.ifm_shapes.append(Shape4D(ifm_tensor.get_full_shape())) self.ofm_shapes.append(Shape4D(ofm_tensor.get_full_shape())) - elif self.type.is_split_op or self.type.is_concat_op(): + elif self.type.is_split_op() or self.type.is_concat_op(): for inp in self.inputs: if inp is not None: self.ifm_shapes.append(Shape4D(full_shape(4, inp.shape, 1))) @@ -751,7 +751,9 @@ class Operation: else: self.ofm_shapes.append(None) else: - self.ifm_shapes.append(Shape4D(full_shape(4, ifm_tensor.shape, 1))) + if ifm_tensor is not None: + self.ifm_shapes.append(Shape4D(full_shape(4, ifm_tensor.shape, 1))) if ifm2_tensor is not None: self.ifm_shapes.append(Shape4D(full_shape(4, ifm2_tensor.shape, 1))) - self.ofm_shapes.append(Shape4D(full_shape(4, ofm_tensor.shape, 1))) + if ofm_tensor is not None: + self.ofm_shapes.append(Shape4D(full_shape(4, ofm_tensor.shape, 1))) |