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/TosaRegion.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/TosaRegion.py')
| -rw-r--r-- | ethosu/vela/tosa/TosaRegion.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/tosa/TosaRegion.py b/ethosu/vela/tosa/TosaRegion.py index 7fd6e3c5..b8a10e32 100644 --- a/ethosu/vela/tosa/TosaRegion.py +++ b/ethosu/vela/tosa/TosaRegion.py @@ -42,7 +42,7 @@ class TosaRegion(object): x = self._tab.Vector(o) x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4 x = self._tab.Indirect(x) - from tosa.TosaBasicBlock import TosaBasicBlock + from .TosaBasicBlock import TosaBasicBlock obj = TosaBasicBlock() obj.Init(self._tab.Bytes, x) return obj |