diff --git a/ethosu/vela/graph_optimiser.py b/ethosu/vela/graph_optimiser.py
index b708b62..f59b685 100644
--- a/ethosu/vela/graph_optimiser.py
+++ b/ethosu/vela/graph_optimiser.py
@@ -263,9 +263,10 @@
     ):
         return
 
-    if not any(cons.run_on_npu for cons in tens.consumer_list):
+    # Check if any of the producers/consumers is run on CPU
+    if not all(cons.run_on_npu for cons in tens.consumer_list):
         return
-    if not any(prod.run_on_npu for prod in tens.ops):
+    if not all(prod.run_on_npu for prod in tens.ops):
         return
 
     # "Concat" ofm exception: