Searched refs:NodePattern (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/torch/ao/quantization/fx/ |
| D | quantize_handler.py | 11 from torch.ao.quantization.utils import NodePattern, Pattern, QuantizerCls
49 node_pattern: NodePattern, argument
115 node_pattern: NodePattern, argument
|
| D | fuse_handler.py | 8 from torch.ao.quantization.utils import _parent_name, NodePattern, Pattern
43 matched_node_pattern: NodePattern, argument
62 matched_node_pattern: NodePattern, argument
|
| D | fuse.py | 14 from torch.ao.quantization.utils import NodePattern, Pattern
140 ) -> Dict[str, Tuple[Node, Pattern, NodePattern, FuseHandler, Dict[Node, Any]]]:
144 str, Tuple[Node, Pattern, NodePattern, FuseHandler, Dict[Node, Any]]
|
| D | prepare.py | 43 NodePattern,
467 def _add_matched_node_name_to_set(matched_node_pattern: NodePattern, s: Set[str]): argument
506 matched_node_pattern: NodePattern, argument
|
| /external/yapf/third_party/yapf_third_party/_ylib2to3/ |
| D | patcomp.py | 167 return pytree.NodePattern(type, content)
|
| D | pytree.py | 555 class NodePattern(BasePattern): class
672 return NodePattern(name=self.name)
|
| /external/pytorch/torch/ao/quantization/ |
| D | utils.py | 17 NodePattern = Union[Tuple[Node, Node], Tuple[Node, Tuple[Node, Node]], Any] variable
18 NodePattern.__module__ = "torch.ao.quantization.utils"
|