diff --git a/backends/arm/operator_support/ethos_u55_support.py b/backends/arm/operator_support/ethos_u55_support.py index 5cd9918b567..574b5aa0bdc 100644 --- a/backends/arm/operator_support/ethos_u55_support.py +++ b/backends/arm/operator_support/ethos_u55_support.py @@ -125,13 +125,13 @@ def is_node_supported( # noqa: C901 ifm_dtype = _try_determine_dtype(ifm) if ifm_dtype is not None and ifm_dtype not in (torch.int8, torch.int16): self.reporter.report_reject( - node, f"Unsupported input dtype {dtype} (Supports i8, i16)." + node, f"Unsupported input dtype {ifm_dtype} (Supports i8, i16)." ) return False weight_dtype = _try_determine_dtype(weight) if weight_dtype is not None and weight_dtype not in (torch.int8,): self.reporter.report_reject( - node, f"Unsupported weight dtype {dtype} (Supports i8)." + node, f"Unsupported weight dtype {weight_dtype} (Supports i8)." ) return False if len(node.all_input_nodes) > 2: @@ -139,7 +139,7 @@ def is_node_supported( # noqa: C901 bias_dtype = _try_determine_dtype(bias) if bias_dtype is not None and bias_dtype not in (torch.int32,): self.reporter.report_reject( - node, f"Unsupported bias dtype {dtype} (Supports i32)." + node, f"Unsupported bias dtype {bias_dtype} (Supports i32)." ) return False