Searched defs:EdgeOrNode (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/ |
| H A D | prepare.py | 36 edge_or_node: EdgeOrNode, shared_with_map: Dict[EdgeOrNode, EdgeOrNode] argument
56 parent: EdgeOrNode,
57 child: EdgeOrNode,
58 shared_with_map: Dict[EdgeOrNode, EdgeOrNode], argument
68 child: EdgeOrNode,
70 shared_with_map: Dict[EdgeOrNode, EdgeOrNode], argument
85 edge_or_node_to_qspec: Dict[EdgeOrNode, QuantizationSpecBase],
86 shared_with_map: Dict[EdgeOrNode, EdgeOrNode], argument
164 edge_or_node_to_qspec: Dict[EdgeOrNode, QuantizationSpecBase]
290 edge_or_node_to_group_id: Dict[EdgeOrNode, int],
[all …]
|
| H A D | qat_utils.py | 597 def _get_new_edge_or_node(edge_or_node: EdgeOrNode):
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/fx/ |
| H A D | prepare.py | 138 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize],
368 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize],
666 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize],
734 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize],
750 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize],
807 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize],
969 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize],
1081 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize],
1162 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize],
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/quantizer/ |
| H A D | quantizer.py | 84 EdgeOrNode = Union[Tuple[Node, Node], Node] variable
|