diff options
author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2021-02-16 21:59:50 +0100 |
---|---|---|
committer | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2021-02-17 09:18:39 +0100 |
commit | 8d0f4890aa0ceae92a33ebb789701ff644a6fcaa (patch) | |
tree | fcecd50a7cc6375f5f4320b42f4b6c5231b854b1 /ethosu/vela/operation.py | |
parent | 56b6c711d8faaa6bcbc810e895efa650ddd97e73 (diff) | |
download | ethos-u-vela-8d0f4890aa0ceae92a33ebb789701ff644a6fcaa.tar.gz |
[MLBEDSW-3813] Fix LSTM operator pass through
Fixed pass through of LSTM operator.
Change-Id: I23140c69ab6cdc83f6bb8129256b4cc6a7c5ffac
Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
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 16431be7..e4d11be5 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -407,6 +407,7 @@ class Operation: "attrs", "inputs", "outputs", + "intermediates", "flops", "scheduled_pass", "run_on_npu", @@ -427,6 +428,7 @@ class Operation: self.attrs: Dict[str, Any] = {} self.inputs: List[Tensor] = [] self.outputs: List[Tensor] = [] + self.intermediates: List[Tensor] = [] self.flops = 0 self.run_on_npu = True # Fused activation function. If not none: operator code. @@ -453,6 +455,7 @@ class Operation: res.attrs = dict(self.attrs) res.inputs = list(self.inputs) res.outputs = list(self.outputs) + res.intermediates = list(self.intermediates) res.flops = self.flops res.run_on_npu = self.run_on_npu res.activation = None if self.activation is None else self.activation.clone() |