Searched refs:ObserverOrFakeQuantize (Results 1 – 6 of 6) sorted by relevance
/external/pytorch/torch/ao/quantization/ |
D | __init__.py | 35 ObserverOrFakeQuantize = Union[ObserverBase, FakeQuantizeBase] variable 36 ObserverOrFakeQuantize.__module__ = "torch.ao.quantization" 192 obs_or_fqs: List[ObserverOrFakeQuantize], argument 194 [List[ObserverOrFakeQuantize]], Tuple[Tensor, Tensor] argument
|
/external/pytorch/torch/ao/quantization/pt2e/ |
D | prepare.py | 9 ObserverOrFakeQuantize, 293 ) -> Dict[EdgeOrNode, ObserverOrFakeQuantize]: 298 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize] = {} 299 group_id_to_obs_or_fq: Dict[int, ObserverOrFakeQuantize] = {} 318 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize], argument 401 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize], argument 451 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize], argument 478 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize], argument
|
/external/pytorch/torch/ao/quantization/fx/ |
D | prepare.py | 14 ObserverOrFakeQuantize, 138 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize], argument 245 obs_or_fq: Optional[ObserverOrFakeQuantize], argument 368 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize], argument 477 obs_or_fq: ObserverOrFakeQuantize, argument 666 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize], argument 668 ) -> ObserverOrFakeQuantize: 734 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize], argument 750 obs_or_fq_map: Dict[EdgeOrNode, ObserverOrFakeQuantize], argument 752 ) -> Optional[ObserverOrFakeQuantize]: [all …]
|
/external/pytorch/torch/ao/quantization/quantizer/ |
D | quantizer.py | 8 from torch.ao.quantization import ObserverOrFakeQuantize 103 derive_qparams_fn: Callable[[List[ObserverOrFakeQuantize]], Tuple[Tensor, Tensor]]
|
/external/executorch/backends/cadence/aot/quantizer/ |
D | utils.py | 17 from torch.ao.quantization import ObserverOrFakeQuantize 128 obs_or_fqs: List[ObserverOrFakeQuantize], argument
|
/external/pytorch/test/quantization/pt2e/ |
D | test_quantize_pt2e.py | 8 from torch.ao.quantization import observer, ObserverOrFakeQuantize, QConfigMapping 339 obs_or_fqs: List[ObserverOrFakeQuantize], argument 442 obs_or_fqs: List[ObserverOrFakeQuantize], argument
|