commit ee99bb124b088430b97d205df9fc90a1e9412e0c
author Patrik Gustavsson <patrik.gustavsson@arm.com> Thu Apr 08 09:04:00 2021 +0200
committer Patrik Gustavsson <patrik.gustavsson@arm.com> Thu Apr 08 11:21:35 2021 +0200
tree bfb1a6799f9e9ff9d0f387471277b8a26edbab70
parent 95b279f1454d58a93238851cb5ff394c7782ad32

MLBEDSW-4334 Non-linear format decision in graph opt.

Check if non linear tensor format can be used is
refactored.

-Flag avoid_NHCWB16 replaced with needs_linear_format
-Checking restrictions located to one function in graph optimiser.

Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: Iec5c7996a1a6039cad052197f1ae56f7c0290440

ethosu/vela/tensor.py

diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py
index e915363..15bd05e 100644
--- a/ethosu/vela/tensor.py
+++ b/ethosu/vela/tensor.py
@@ -372,7 +372,7 @@
         "block_traversal",
         "equivalence_id",
         "resampling_mode",
-        "avoid_NHCWB16",
+        "needs_linear_format",
     )
     AllocationQuantum = 16
 
@@ -418,7 +418,7 @@
         self.block_traversal: TensorBlockTraversal = TensorBlockTraversal.Default
         self.resampling_mode: resampling_mode = resampling_mode.NONE
 
-        self.avoid_NHCWB16: bool = False
+        self.needs_linear_format = True
 
     @property
     def address(self) -> int: