Searched refs:QuantizationSpecBase (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/torch/ao/quantization/quantizer/ |
| D | quantizer.py | 25 class QuantizationSpecBase(ABC): # noqa: B024 class
32 class QuantizationSpec(QuantizationSpecBase):
68 class FixedQParamsQuantizationSpec(QuantizationSpecBase):
89 class SharedQuantizationSpec(QuantizationSpecBase):
99 class DerivedQuantizationSpec(QuantizationSpecBase):
120 input_qspec_map: Dict[Node, Optional[QuantizationSpecBase]] = field(
126 output_qspec: Optional[QuantizationSpecBase] = None
|
| D | __init__.py | 7 QuantizationSpecBase,
|
| D | xnnpack_quantizer_utils.py | 22 QuantizationSpecBase,
602 input_qspec_map: Dict[Node, QuantizationSpecBase] = {}
|
| /external/pytorch/torch/ao/quantization/pt2e/ |
| D | prepare.py | 22 QuantizationSpecBase,
69 qspec: QuantizationSpecBase, argument
84 qspec: QuantizationSpecBase, argument
85 edge_or_node_to_qspec: Dict[EdgeOrNode, QuantizationSpecBase], argument
87 ) -> QuantizationSpecBase:
101 def _has_same_dtype(qspec_a: QuantizationSpecBase, qspec_b: QuantizationSpecBase): argument
109 def _has_same_is_dynamic(qspec_a: QuantizationSpecBase, qspec_b: QuantizationSpecBase): argument
119 ) -> Dict[EdgeOrNode, QuantizationSpecBase]:
121 edge_or_node_to_qspec: Dict[EdgeOrNode, QuantizationSpecBase] = {}
164 edge_or_node_to_qspec: Dict[EdgeOrNode, QuantizationSpecBase] argument
[all …]
|
| D | port_metadata_pass.py | 12 from torch.ao.quantization.quantizer import QuantizationSpecBase
74 qspec: Optional[QuantizationSpecBase], argument
132 node: torch.fx.Node, qspec: Optional[QuantizationSpecBase] argument
|
| D | qat_utils.py | 14 QuantizationSpecBase,
614 def _get_new_qspec(qspec: QuantizationSpecBase): argument
|
| /external/pytorch/torch/ao/quantization/ |
| D | quantize_pt2e.py | 10 QuantizationSpecBase,
|
| /external/pytorch/torch/ao/quantization/fx/ |
| D | prepare.py | 36 QuantizationSpecBase,
128 input_qspec_map: Dict[Node, QuantizationSpecBase], argument
130 ) -> Optional[QuantizationSpecBase]:
137 quantization_spec: Optional[QuantizationSpecBase], argument
|