diff options
author | Jacob Bohlin <jacob.bohlin@arm.com> | 2020-08-17 10:13:04 +0200 |
---|---|---|
committer | tim.hall <tim.hall@arm.com> | 2020-08-18 18:53:03 +0000 |
commit | 30cb47abd68b125d5c3fc315948187a9ea8dbd43 (patch) | |
tree | adbf962277fefcfa0c0d5beda5a9d769b123d01f | |
parent | 940fdee086d77089c7669190392c92f4f2150348 (diff) | |
download | ethos-u-vela-30cb47abd68b125d5c3fc315948187a9ea8dbd43.tar.gz |
MLBEDSW-2718: Fix a bug that would reshape TransposeConv input
Signed-off-by: Jacob Bohlin <jacob.bohlin@arm.com>
Change-Id: If22fd21f9953a62305620a4e804e5caacb342c89
-rw-r--r-- | ethosu/vela/tflite_reader.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ethosu/vela/tflite_reader.py b/ethosu/vela/tflite_reader.py index 5e966d1b..daa208f1 100644 --- a/ethosu/vela/tflite_reader.py +++ b/ethosu/vela/tflite_reader.py @@ -153,7 +153,8 @@ class TFLiteSubgraph: if op_type.startswith("DepthwiseConv2d") or op_type.startswith("Conv2D"): inputs[1] = clone_and_reshape_tensor(inputs[1], (1, 2, 3, 0)) - inputs[2] = clone_and_reshape_tensor(inputs[2], (0,)) + if not op.type.endswith("BackpropInput"): + inputs[2] = clone_and_reshape_tensor(inputs[2], (0,)) if op_type.startswith("FullyConnected"): inputs[1] = clone_and_reshape_tensor(inputs[1], (1, 0)) |