NXP backend: Fix batch norm tests.

Author: MartinPavella

backends/nxp/tests/test_batch_norm_fusion.py

@@ -105,8 +105,8 @@ def test_batch_norm_conv_fusing(bias: bool, input_shape: list[int]):
     og_nodes = list(program.graph.nodes)
     transformed_nodes = list(graph_module_out.graph.nodes)
 
-    assert len(og_nodes) == (11 if bias else 10)
-    assert og_nodes[9 if bias else 8].target.__name__ == "batch_norm.default"
+    assert len(og_nodes) == (10 if bias else 9)
+    assert og_nodes[8 if bias else 7].target.__name__ == "batch_norm.default"
 
     assert len(transformed_nodes) == 5
     assert not any(
@@ -139,8 +139,8 @@ def test_batch_norm_linear_fusing(bias: bool):
     og_nodes = list(og_module.graph.nodes)
     transformed_nodes = list(graph_module_out.graph.nodes)
 
-    assert len(og_nodes) == (11 if bias else 10)
-    assert og_nodes[8 if bias else 7].target.__name__ == "linear.default"
+    assert len(og_nodes) == (10 if bias else 9)
+    assert og_nodes[7 if bias else 6].target.__name__ == "linear.default"
 
     assert len(transformed_nodes) == 5
     assert not any(