We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8ce0efd commit 6fac514Copy full SHA for 6fac514
backends/nxp/quantizer/patterns.py
@@ -231,11 +231,11 @@ class SubTensorPattern(QuantizationPattern):
231
Basic quantization for all inputs and output.
232
"""
233
234
- def partition_types(self) -> List[Type[torch.nn.Module]]:
+ def partition_types(self) -> list[type[torch.nn.Module]]:
235
return [torch.ops.aten.sub.Tensor]
236
237
def get_anchors(
238
- self, gm: fx.GraphModule, fused_partition: List[fx.GraphModule]
+ self, gm: fx.GraphModule, fused_partition: list[fx.GraphModule]
239
) -> PartitionAnchors | None:
240
node = fused_partition[0].nodes[-1]
241
inputs = [(node, 0)]
0 commit comments