Fix upsteam review comments

Signed-off-by: Adrian Lundell <[email protected]>

Author: AdrianLundell

backends/arm/_passes/to_tosa_memory_format_pass.py

@@ -253,22 +253,22 @@ def insert_tosa_transposes(self, graph_module: torch.fx.GraphModule):
         for input_node in inputs:
             input_dim_order = get_first_fake_tensor(input_node).dim_order()
             if input_dim_order in (NCHW_ORDER, NNCHW_ORDER):
-                ToTosaMemoryFormatPass.insert_output_transpose(input_node, graph_module)
+                self.insert_output_transpose(input_node, graph_module)
 
         # Transpose outputs if they are in (N)NCHW format
         outputs = output_node.args[0]
         output_dim_orders = output_node.meta.get("original_dim_orders")
         if output_dim_orders is None:
             raise RuntimeError(
-                f"{AnnotateDecomposedMatmulPass.__name__} is required to run at the beginning of the pass pipeline when using {ToTosaMemoryFormatPass.__name__}."
+                f"{AnnotateDecomposedMatmulPass.__name__} is required to run at the beginning of the pass pipeline when using {self.__name__}."
             )
 
         for output_node_input, output_dim_order in zip(outputs, output_dim_orders):  # type: ignore[arg-type]
             if output_dim_order in (
                 NCHW_ORDER,
                 NNCHW_ORDER,
             ):
-                ToTosaMemoryFormatPass.insert_input_transpose(
+                self.insert_input_transpose(
                     output_node, output_node_input, graph_module
                 )
 

backends/arm/test/targets.bzl

@@ -39,7 +39,7 @@ def define_arm_tests():
         "misc/test_bn_relu_folding_qat.py",
         "misc/test_custom_partition.py",
         "misc/test_debug_hook.py",
-        "misc/test_dim_order_guards.py",
+        "misc/test_dim_order.py",
         "misc/test_outputs_order.py",
     ]