diff options
author | Jacob Bohlin <jacob.bohlin@arm.com> | 2020-08-24 11:42:14 +0200 |
---|---|---|
committer | Jacob Bohlin <jacob.bohlin@arm.com> | 2020-08-26 08:57:59 +0200 |
commit | d47cc2777f3a3f371958c30a3c1880f692b0b0a2 (patch) | |
tree | 47dd9c57f13719dbbc6bd886042f9c8795353c98 /ethosu/vela/graph_optimiser.py | |
parent | 814cfbb8124ba0b3828db2bb12d9342ae9c39f19 (diff) | |
download | ethos-u-vela-d47cc2777f3a3f371958c30a3c1880f692b0b0a2.tar.gz |
MLBEDSW-2847: Fix for TransposeConv crash and u8 output diff
Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: I2cb3f6639e4bb8a984fa3647ee7b4678ed6f5890
Diffstat (limited to 'ethosu/vela/graph_optimiser.py')
-rw-r--r-- | ethosu/vela/graph_optimiser.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py index 9f92e755..b9aafcac 100644 --- a/ethosu/vela/graph_optimiser.py +++ b/ethosu/vela/graph_optimiser.py @@ -156,8 +156,8 @@ def calc_upscaled_padding_and_skirt(padding_type, kernel_size, stride, input_dim 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_attrs_to_resizebilinear(op, arch): 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] |