diff options
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 |