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/tensor.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/tensor.py')
| -rw-r--r-- | ethosu/vela/tensor.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index b7d4307f..97885d0e 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -341,6 +341,7 @@ class Tensor: "bandwidth_shape", "dtype", "name", + "is_variable", "ops", "consumer_list", "values", @@ -378,6 +379,7 @@ class Tensor: self.bandwidth_shape = shape self.dtype = dtype self.name = name + self.is_variable = False self.equivalence_id: UUID = uuid.uuid4() self.ops: List[Operation] = [] |