diff options
author | Louis Verhaard <louis.verhaard@arm.com> | 2021-01-20 17:23:54 +0100 |
---|---|---|
committer | Louis Verhaard <louis.verhaard@arm.com> | 2021-02-12 17:40:21 +0100 |
commit | d70025250fc49997801ea3a6ce83f2fa29a09d78 (patch) | |
tree | 07462a32ed30ba9893bb7825e44b9606a400e709 /ethosu/vela/nn_graph.py | |
parent | ad45f792e699fe6abdc381f62690801aa50bd412 (diff) | |
download | ethos-u-vela-d70025250fc49997801ea3a6ce83f2fa29a09d78.tar.gz |
MLBEDSW-3808: Ported search allocator to python2.1.0.rc1
- 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>
Diffstat (limited to 'ethosu/vela/nn_graph.py')
-rw-r--r-- | ethosu/vela/nn_graph.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py index db878bc3..c45d0e3e 100644 --- a/ethosu/vela/nn_graph.py +++ b/ethosu/vela/nn_graph.py @@ -38,7 +38,7 @@ class PassPlacement(enum.Enum): class TensorAllocator(enum.Enum): LinearAlloc = 1 Greedy = 2 - Search = 3 + HillClimb = 3 def __str__(self): return self.name |