diff options
| -rw-r--r-- | ethosu/vela/tflite_graph_optimiser.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/tflite_graph_optimiser.py b/ethosu/vela/tflite_graph_optimiser.py index ae11bec..85fb8ba 100644 --- a/ethosu/vela/tflite_graph_optimiser.py +++ b/ethosu/vela/tflite_graph_optimiser.py @@ -1743,6 +1743,7 @@ def fuse_activation_function_with_prev(op, arch, nng): # Note: the below checks on prev_op require that a first optimize pass on the full graph has been performed fuse = ( prev_op.run_on_npu + and prev_op.type != Op.Memcpy and prev_op.type.npu_block_type != NpuBlockType.Default and len(ifm.ops) == 1 and len(prev_op.outputs[0].consumers()) == 1 |