aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/high_level_command_stream_generator.py
diff options
context:
space:
mode:
authorJacob Bohlin <jacob.bohlin@arm.com>2020-07-07 17:15:22 +0200
committerJacob Bohlin <jacob.bohlin@arm.com>2020-07-13 12:21:01 +0200
commit9b64ba0f5b45359db4c12fa77fc2caffb09f4432 (patch)
tree94d7ea3e21175614005332430240460593f52de9 /ethosu/vela/high_level_command_stream_generator.py
parent9fbc4913fe5dd0e6090532963f8612449936d994 (diff)
downloadethos-u-vela-9b64ba0f5b45359db4c12fa77fc2caffb09f4432.tar.gz
MLBEDSW-2584: Support cascading of Transpose Convolution
Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com> Change-Id: I39cff126dda89d71426ab731427ca1d64d02590d
Diffstat (limited to 'ethosu/vela/high_level_command_stream_generator.py')
-rw-r--r--ethosu/vela/high_level_command_stream_generator.py1
1 files changed, 0 insertions, 1 deletions
diff --git a/ethosu/vela/high_level_command_stream_generator.py b/ethosu/vela/high_level_command_stream_generator.py
index 232a56c1..6aa88d86 100644
--- a/ethosu/vela/high_level_command_stream_generator.py
+++ b/ethosu/vela/high_level_command_stream_generator.py
@@ -79,7 +79,6 @@ def generate_high_level_command_stream_for_pass(strat, passes, block_configs, id
skirt = ps.primary_op.attrs.get("skirt", None)
if ps.primary_op.type in set(("Conv2DBackpropInputSwitchedBias", "ResizeBilinear")):
upscaling = ofm_tensor.shape[-3] // ifm_tensor.shape[-3]
- assert ofm_tensor.shape[-2] == (ifm_tensor.shape[-2] * upscaling)
concat_axis = 0
concat_offset = 0