From e3b1b91c450129308a3a1d466a2f2876a5a872b1 Mon Sep 17 00:00:00 2001 From: Patrik Gustavsson Date: Tue, 9 Feb 2021 15:38:46 +0100 Subject: MLBEDSW-3774 Remove SplitSliceRead Removed SplitSliceRead from subgraph during graph optimisation. Signed-off-by: Patrik Gustavsson Change-Id: I9315d4c2a6767828dd2b4e66823d73b10ebee99c --- ethosu/vela/operation.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'ethosu/vela/operation.py') 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 -- cgit v1.2.1