Fix Transpose Optimization Bug With non-4D Tensor Input

Summary: Fixed a bug in the channels last tagged reshape pass, where non-4d inputs were being tagged as contiguous/channels last memory formats, which isn't expected as these formats only apply to 4d tensors. The repro is in N7569847. The fix was completed by checking tensor shape size before tagging input nodes.

Reviewed By: mcr229

Differential Revision: D78357428

Author: leafs1

backends/xnnpack/_passes/channels_last_tagged_reshape_pass.py

@@ -457,7 +457,7 @@ def call(self, graph_module: torch.fx.GraphModule):  # noqa: C901
         for node in original_nodes:
             if len(node.all_input_nodes) == 0:
                 # This node has no inputs so we don't need to change anything, but still need to tag input nodes
-                if "val" in node.meta and isinstance(node.meta["val"], torch.Tensor):
+                if "val" in node.meta and isinstance(node.meta["val"], torch.Tensor) and len(node.meta["val"].shape) == 4:
                     if node.meta["val"].is_contiguous():
                         self.mark_as_nchw_node(node)
                     else: