Searched defs:QuantizationSpecBase (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/ |
| H A D | prepare.py | 69 qspec: QuantizationSpecBase,
84 qspec: QuantizationSpecBase,
85 edge_or_node_to_qspec: Dict[EdgeOrNode, QuantizationSpecBase], argument
101 def _has_same_dtype(qspec_a: QuantizationSpecBase, qspec_b: QuantizationSpecBase):
109 def _has_same_is_dynamic(qspec_a: QuantizationSpecBase, qspec_b: QuantizationSpecBase):
164 edge_or_node_to_qspec: Dict[EdgeOrNode, QuantizationSpecBase] argument
291 edge_or_node_to_qspec: Dict[EdgeOrNode, QuantizationSpecBase], argument
|
| H A D | port_metadata_pass.py | 74 qspec: Optional[QuantizationSpecBase],
132 node: torch.fx.Node, qspec: Optional[QuantizationSpecBase]
|
| H A D | qat_utils.py | 614 def _get_new_qspec(qspec: QuantizationSpecBase):
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/quantizer/ |
| H A D | quantizer.py | 25 class QuantizationSpecBase(ABC): # noqa: B024 class
|
| /aosp_15_r20/external/pytorch/torch/ao/quantization/fx/ |
| H A D | prepare.py | 128 input_qspec_map: Dict[Node, QuantizationSpecBase], argument
137 quantization_spec: Optional[QuantizationSpecBase],
|