commit d47cc2777f3a3f371958c30a3c1880f692b0b0a2
author Jacob Bohlin <jacob.bohlin@arm.com> Mon Aug 24 11:42:14 2020 +0200
committer Jacob Bohlin <jacob.bohlin@arm.com> Wed Aug 26 08:57:59 2020 +0200
tree 47dd9c57f13719dbbc6bd886042f9c8795353c98
parent 814cfbb8124ba0b3828db2bb12d9342ae9c39f19

MLBEDSW-2847: Fix for TransposeConv crash and u8 output diff

Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: I2cb3f6639e4bb8a984fa3647ee7b4678ed6f5890

ethosu/vela/graph_optimiser.py

diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py
index 9f92e75..b9aafca 100644
--- a/ethosu/vela/graph_optimiser.py
+++ b/ethosu/vela/graph_optimiser.py
@@ -156,8 +156,8 @@
         ypad = needed_total_padding(int(input_dims[1]) * upscaling_factor, int(stride[1]), int(kernel_height))
         xpad = needed_total_padding(int(input_dims[2]) * upscaling_factor, int(stride[2]), int(kernel_width))
 
-        right_pad = ((xpad + 1) // upscaling_factor) - 1
-        bottom_pad = ((ypad + 1) // upscaling_factor) - 1
+        right_pad = max(((xpad + 1) // upscaling_factor) - 1, 0)
+        bottom_pad = max(((ypad + 1) // upscaling_factor) - 1, 0)
         left_pad = max(kernel_width - 1 - right_pad, 0)
         top_pad = max(kernel_height - 1 - bottom_pad, 0)
 
@@ -845,7 +845,7 @@
 
 
 def add_bias_tensor(op, arch):
-    if ("Conv2d" in op.type or op.type.startswith("FullyConnected")) and not op.inputs[-1]:
+    if ("conv2d" in op.type.lower() or op.type.startswith("FullyConnected")) and not op.inputs[-1]:
         # Add bias/scale tensor filled with zeros
         weight_shape = op.inputs[1].shape
         weight_sets = weight_shape[-1]