commit e3d18b0a77caa20ccc0c32fbaba6ff1564e4476d
author Diqing Zhong <diqing.zhong@arm.com> Mon Nov 15 13:53:10 2021 +0100
committer Diqing Zhong <diqing.zhong@arm.com> Mon Nov 15 14:21:15 2021 +0100
tree d0fd4395066692f17a3602688676cb5e141cbd73
parent 399c4a2d77df791e5d988c51d7fb1824ac4f266f

TOSA: Add ifm ofm elem size into raw output

Change-Id: I645496536a6bddf2bd289a87be9d7cef11693954
Signed-off-by: Diqing Zhong <diqing.zhong@arm.com>

ethosu/vela/tensor.py

diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py
index 8c5e277..ab63afd 100644
--- a/ethosu/vela/tensor.py
+++ b/ethosu/vela/tensor.py
@@ -434,7 +434,7 @@
 
     def element_size(self) -> int:
         if self.element_size_bytes == 0:
-            return self.dtype.size_in_bits() / 8
+            return self.dtype.size_in_bits() // 8
         return self.element_size_bytes
 
     # Returns a copy, renamed to self.name + suffix