commit a0c3624899edc601525a589643c802469003f89d
author Fredrik Svedberg <fredrik.svedberg@arm.com> Wed Jun 03 15:43:31 2020 +0200
committer Louis Verhaard <louis.verhaard@arm.com> Wed Aug 05 16:26:04 2020 +0200
tree 4fc52db04cd29901b3e5d4a7425a7a641e9647fb
parent 9a03fdff316662be69a1adc4e391e43bc6519b08

[MLBEDSW-2335] SoftMax int16

Added graph rewrite of Softmax for int16.

Change-Id: Id7885af6056a23e8b8362fb61ae94283251eb398
Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>

ethosu/vela/nn_graph.py

diff --git a/ethosu/vela/nn_graph.py b/ethosu/vela/nn_graph.py
index 1024307..4a2855b 100644
--- a/ethosu/vela/nn_graph.py
+++ b/ethosu/vela/nn_graph.py
@@ -51,6 +51,7 @@
         self.ofm_tensor = None
         self.weight_tensor = None
         self.scale_tensor = None
+        self.lut_tensor = None
         self.name = name
         self.cascade = None
         self.placement = placement
@@ -85,6 +86,11 @@
             return None, None, None, None
         return self.primary_op.get_ifm_weights_biases_ofm()
 
+    def get_primary_op_lut(self):
+        if not self.primary_op:
+            return None
+        return self.primary_op.activation_lut
+
 
 class SchedulingStrategy(enum.Enum):
     Unknown = -1