(node: torch.fx.Node)
| 93 | |
| 94 | |
| 95 | def is_quant(node: torch.fx.Node) -> bool: |
| 96 | if node.op != "call_function": |
| 97 | return False |
| 98 | node_name = format_target_name(node.target.__name__) # pyre-ignore |
| 99 | |
| 100 | return node_name in _Q_OPS |
| 101 | |
| 102 | |
| 103 | def is_dequant(node: torch.fx.Node) -> bool: |