commit d70025250fc49997801ea3a6ce83f2fa29a09d78
author Louis Verhaard <louis.verhaard@arm.com> Wed Jan 20 17:23:54 2021 +0100
committer Louis Verhaard <louis.verhaard@arm.com> Fri Feb 12 17:40:21 2021 +0100
tree 07462a32ed30ba9893bb7825e44b9606a400e709
parent ad45f792e699fe6abdc381f62690801aa50bd412

MLBEDSW-3808: Ported search allocator to python

- Straight port of the C++ implementation to python.
- Renamed the allocator from "Search" to "HillClimb"

Change-Id: I50797d541f326d0264daf79bf7866aef32350a60
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>

ethosu/vela/nn_graph.py

diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py
index db878bc..c45d0e3 100644
--- a/ethosu/vela/nn_graph.py
+++ b/ethosu/vela/nn_graph.py
@@ -38,7 +38,7 @@
 class TensorAllocator(enum.Enum):
     LinearAlloc = 1
     Greedy = 2
-    Search = 3
+    HillClimb = 3
 
     def __str__(self):
         return self.name