Skip to content

Commit ae15253

Browse files
authored
Don't Decompose Hardswish
Differential Revision: D77765129 Pull Request resolved: #12360
1 parent 5c364d9 commit ae15253

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

backends/xnnpack/partition/config/generic_node_configs.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -400,6 +400,9 @@ class HardswishConfig(GenericNodePartitionerConfig):
400400
def supported_precision_types(self) -> List[ConfigPrecisionType]:
401401
return [ConfigPrecisionType.FP32]
402402

403+
def get_original_aten(self) -> Optional[torch._ops.OpOverload]:
404+
return torch.ops.aten.hardswish.default
405+
403406

404407
class LeakyReLUConfig(GenericNodePartitionerConfig):
405408
target_name = "leaky_relu.default"

0 commit comments

Comments
 (0)