diff options
| author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-02-09 15:38:46 +0100 |
|---|---|---|
| committer | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2021-02-11 14:38:56 +0100 |
| commit | e3b1b91c450129308a3a1d466a2f2876a5a872b1 (patch) | |
| tree | 77c6fe5cc8836dc3aac505efdbf78cee433bd398 /ethosu/vela/operation.py | |
| parent | 138d47f5a3e87d294b3714ae799ccad8ac9111bd (diff) | |
| download | ethos-u-vela-e3b1b91c450129308a3a1d466a2f2876a5a872b1.tar.gz | |
MLBEDSW-3774 Remove SplitSliceRead
Removed SplitSliceRead from subgraph during
graph optimisation.
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: I9315d4c2a6767828dd2b4e66823d73b10ebee99c
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 b297bed0..16431be7 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -418,6 +418,7 @@ class Operation: "ifm_shapes", "ofm_shapes", "rescale", + "read_offsets", ) def __init__(self, op_type: Op, name: str): @@ -444,6 +445,7 @@ class Operation: # If not none: contains rescale to be used as output scaling # (which overrides the ofm tensor's scale) self.rescale = None + self.read_offsets: List[Shape4D] = [None, None] # offset for [ifm, ifm2] def clone(self, suffix="_clone"): res = Operation(self.type, self.name + suffix) @@ -458,6 +460,7 @@ class Operation: res.forced_output_quantization = self.forced_output_quantization res.scheduled_pass = self.scheduled_pass res.op_index = None # not relevant as not part of input network + res.read_offsets = list(self.read_offsets) return res |