diff options
author | Kevin Cheng <kevin.cheng@arm.com> | 2021-09-02 13:43:17 -0700 |
---|---|---|
committer | Kevin Cheng <kevin.cheng@arm.com> | 2021-09-14 14:54:51 -0700 |
commit | b227ae5ff22411c3f7c82e2b4b5a4618008e897a (patch) | |
tree | 71072b49d2b1d102634f454e536b9aa7144a899d /verif/tosa_serializer.py | |
parent | 9034747c223cd5ab53148ec40aaf64765fcd6531 (diff) | |
download | reference_model-b227ae5ff22411c3f7c82e2b4b5a4618008e897a.tar.gz |
Fix while_loop test.
- add output tensor to main block
- fix bug where tensor cannot serialize list of const data
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: I40b96d634642e24e6fae6fa3ad6846f5911005f7
Diffstat (limited to 'verif/tosa_serializer.py')
-rw-r--r-- | verif/tosa_serializer.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/verif/tosa_serializer.py b/verif/tosa_serializer.py index b7f5565..e9ec027 100644 --- a/verif/tosa_serializer.py +++ b/verif/tosa_serializer.py @@ -358,6 +358,9 @@ class TosaSerializerTensor: data = data.flatten().astype(int).tolist() data = list(map(int, data)) self.data = data + elif isinstance(data, list): + data = list(map(int, data)) + self.data = data else: self.data = None |