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 5c364d9 commit ae15253Copy full SHA for ae15253
backends/xnnpack/partition/config/generic_node_configs.py
@@ -400,6 +400,9 @@ class HardswishConfig(GenericNodePartitionerConfig):
400
def supported_precision_types(self) -> List[ConfigPrecisionType]:
401
return [ConfigPrecisionType.FP32]
402
403
+ def get_original_aten(self) -> Optional[torch._ops.OpOverload]:
404
+ return torch.ops.aten.hardswish.default
405
+
406
407
class LeakyReLUConfig(GenericNodePartitionerConfig):
408
target_name = "leaky_relu.default"
0 commit comments