diff options
author | Oscar Andersson <oscar.andersson@arm.com> | 2024-02-29 14:35:58 +0100 |
---|---|---|
committer | oscarandersson8218 <oscar.andersson@arm.com> | 2024-03-07 15:54:54 +0100 |
commit | b90666d9b43f4b5223bb4dcecdbee87b2ad757c2 (patch) | |
tree | f4f3ce9c1a6af4625f68a21e68139140a4dd2119 /ethosu/vela/tosa/TosaBasicBlock.py | |
parent | 9341bf4b8de7b5a4db72281490492f6e32b0c605 (diff) | |
download | ethos-u-vela-b90666d9b43f4b5223bb4dcecdbee87b2ad757c2.tar.gz |
TOSA fixes
- Fix TOSA imports
- Handle weights connected to Identity nodes
- Scaling info was missing in Fully Connected
- Disable rescaling fusing for conv-like ops
- Explicit scaling was missing for conv-like ops
- Handle Const->Identity->Transpose chains
- Handle Const->Identity->Reshape chains
Change-Id: I063af1f187b6b56105ccf5e8e8b2eb0d3a39dd3b
Signed-off-by: Oscar Andersson <oscar.andersson@arm.com>
Diffstat (limited to 'ethosu/vela/tosa/TosaBasicBlock.py')
-rw-r--r-- | ethosu/vela/tosa/TosaBasicBlock.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/tosa/TosaBasicBlock.py b/ethosu/vela/tosa/TosaBasicBlock.py index b31f455f..e003a81d 100644 --- a/ethosu/vela/tosa/TosaBasicBlock.py +++ b/ethosu/vela/tosa/TosaBasicBlock.py @@ -42,7 +42,7 @@ class TosaBasicBlock(object): x = self._tab.Vector(o) x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4 x = self._tab.Indirect(x) - from tosa.TosaOperator import TosaOperator + from .TosaOperator import TosaOperator obj = TosaOperator() obj.Init(self._tab.Bytes, x) return obj @@ -67,7 +67,7 @@ class TosaBasicBlock(object): x = self._tab.Vector(o) x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4 x = self._tab.Indirect(x) - from tosa.TosaTensor import TosaTensor + from .TosaTensor import TosaTensor obj = TosaTensor() obj.Init(self._tab.Bytes, x) return obj |