diff options
author | Tim Hall <tim.hall@arm.com> | 2022-03-16 16:31:57 +0000 |
---|---|---|
committer | tim.hall <tim.hall@arm.com> | 2022-03-21 18:50:48 +0000 |
commit | 3c5cfe9e110b402f60fa7e1cdd5aa5e1c31bd511 (patch) | |
tree | c5db3f145a097293cacaaee4112cc30be3670b4c /ethosu/vela/operation.py | |
parent | 845e23200d471e44f274940846e400d170b5ff37 (diff) | |
download | ethos-u-vela-3c5cfe9e110b402f60fa7e1cdd5aa5e1c31bd511.tar.gz |
MLBEDSW-6298: MLCE: Unable to find a valid block config
- Fixed a bug due to ResizeBilinear modifying the attributes of a
shared IFM
- The ifm_resampling_mode is now an attribute of an operator rather
than a tensor
- Changed all calls to try_block_config() to use the attribute rather
than recalculating it in multiple places
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I4641e9cd6b049bd4186776d98e3e751c5e5bcc06
Diffstat (limited to 'ethosu/vela/operation.py')
-rw-r--r-- | ethosu/vela/operation.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 5a6423d8..d5b92d8d 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -31,6 +31,7 @@ from typing import Union from .api import NpuRoundingMode from .errors import VelaError +from .ethos_u55_regs.ethos_u55_regs import resampling_mode from .numeric_util import full_shape from .shape4d import Shape4D @@ -488,6 +489,7 @@ class Operation: "low_precision_scaling", "write_offset", "write_shape", + "ifm_resampling_mode", ) def __init__(self, op_type: Op, name: str): @@ -530,6 +532,7 @@ class Operation: # E.g. an operation that only fills the bottom row of an OFM of size 1x10x8x1 would have # write_offset 0,9,0,0, write_shape 1,1,8,1 self.write_shape: Optional[Shape4D] = None + self.ifm_resampling_mode: resampling_mode = resampling_mode.NONE def clone(self, suffix="_clone"): res = Operation(self.type, self.name + suffix) |