diff options
| author | Louis Verhaard <louis.verhaard@arm.com> | 2020-11-25 20:58:51 +0100 |
|---|---|---|
| committer | Louis Verhaard <louis.verhaard@arm.com> | 2020-11-26 08:57:21 +0100 |
| commit | c21d1c76b21f337d46990c2c262bda536f15e930 (patch) | |
| tree | efba6330962b56850d7b62c5acdec43365c56636 /ethosu/vela/graph_optimiser.py | |
| parent | 0dd21c79ac6ef588e23393064d25e402e16cc2dd (diff) | |
| download | ethos-u-vela-c21d1c76b21f337d46990c2c262bda536f15e930.tar.gz | |
MLBEDSW-3583: Prevent DMA for converted FullyConnected
Do not use DMA for weights of a FullyConnected op that has
been converted to a Conv2D.
Change-Id: Ibf6710c0a1723c8b48c563ca204f274af5ca88ce
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
Diffstat (limited to 'ethosu/vela/graph_optimiser.py')
| -rw-r--r-- | ethosu/vela/graph_optimiser.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py index 58899054..85c2b602 100644 --- a/ethosu/vela/graph_optimiser.py +++ b/ethosu/vela/graph_optimiser.py @@ -338,6 +338,7 @@ def convert_batched_fc_to_conv(op, arch, nng): "stride_h": 1, "stride_w": 1, "strides": (1, 1, 1, 1), + "is_converted_fc": True, } prev_op = ifm.ops[0] |