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/TosaGraph.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/TosaGraph.py')
| -rw-r--r-- | ethosu/vela/tosa/TosaGraph.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/tosa/TosaGraph.py b/ethosu/vela/tosa/TosaGraph.py index 84b51a78..70680561 100644 --- a/ethosu/vela/tosa/TosaGraph.py +++ b/ethosu/vela/tosa/TosaGraph.py @@ -33,7 +33,7 @@ class TosaGraph(object): o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4)) if o != 0: x = self._tab.Indirect(o + self._tab.Pos) - from tosa.Version import Version + from .Version import Version obj = Version() obj.Init(self._tab.Bytes, x) return obj @@ -46,7 +46,7 @@ class TosaGraph(object): x = self._tab.Vector(o) x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4 x = self._tab.Indirect(x) - from tosa.TosaRegion import TosaRegion + from .TosaRegion import TosaRegion obj = TosaRegion() obj.Init(self._tab.Bytes, x) return obj |