diff options
| author | Diqing Zhong <diqing.zhong@arm.com> | 2021-11-15 13:53:10 +0100 |
|---|---|---|
| committer | Diqing Zhong <diqing.zhong@arm.com> | 2021-11-15 14:21:15 +0100 |
| commit | e3d18b0a77caa20ccc0c32fbaba6ff1564e4476d (patch) | |
| tree | d0fd4395066692f17a3602688676cb5e141cbd73 /ethosu/vela/tensor.py | |
| parent | 399c4a2d77df791e5d988c51d7fb1824ac4f266f (diff) | |
| download | ethos-u-vela-3.2.0.rc2.tar.gz | |
TOSA: Add ifm ofm elem size into raw output3.2.0.rc2
Change-Id: I645496536a6bddf2bd289a87be9d7cef11693954
Signed-off-by: Diqing Zhong <diqing.zhong@arm.com>
Diffstat (limited to 'ethosu/vela/tensor.py')
| -rw-r--r-- | ethosu/vela/tensor.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 8c5e277a..ab63afd8 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -434,7 +434,7 @@ class Tensor: 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 |