Arm backend: Add __repr__ to TosaArg

Signed-off-by: Per Åstrand <[email protected]>
Change-Id: I807271caffb3d70641f22d0dfc4835060aa069b2

Author: per

backends/arm/tosa_mapping.py

@@ -79,22 +79,16 @@ def extract_tensor_meta(meta):
 # Class to capture arguments and turn into tensor references for TOSA OPs
 class TosaArg:
     def __process_node(self, argument: torch.fx.Node):
-        self.name = argument.name
+        self.name: str = argument.name
         self.dtype, self.shape, self.dim_order = extract_tensor_meta(argument.meta)
 
     def __process_list(self, argument):
-        self.special = list(argument)
+        self.special: list = list(argument)
 
     def __process_number(self, argument: float | int):
-        self.number = argument
+        self.number: float | int = argument
 
     def __init__(self, argument: Any) -> None:
-        self.name = None  # type: ignore[assignment]
-        self.dtype = None
-        self.shape = None
-        self.dim_order = None
-        self.special = None
-
         if argument is None:
             return
 
@@ -114,3 +108,20 @@ def __init__(self, argument: Any) -> None:
         raise RuntimeError(
             f"Unhandled node input argument: {argument}, of type {type(argument)}"
         )
+
+    def __repr__(self):
+        attrs = []
+        if hasattr(self, "name"):
+            if self.name is not None:
+                attrs.append(f"name={self.name!r}")
+            if self.dtype is not None:
+                attrs.append(f"dtype={ts.DTypeNames[self.dtype]}")
+            if self.shape is not None:
+                attrs.append(f"shape={self.shape!r}")
+            if self.dim_order is not None:
+                attrs.append(f"dim_order={self.dim_order!r}")
+        if hasattr(self, "special") and self.special is not None:
+            attrs.append(f"special={self.special!r}")
+        if hasattr(self, "number") and self.number is not None:
+            attrs.append(f"number={self.number!r}")
+        return f"{self.__class__.__name__}({', '.join(attrs)})"