Home
last modified time | relevance | path

Searched refs:ObserverOrFakeQuantize (Results 1 – 6 of 6) sorted by relevance

/external/pytorch/torch/ao/quantization/
D__init__.py35 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/
Dprepare.py9 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/
Dprepare.py14 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/
Dquantizer.py8 from torch.ao.quantization import ObserverOrFakeQuantize
103 derive_qparams_fn: Callable[[List[ObserverOrFakeQuantize]], Tuple[Tensor, Tensor]]
/external/executorch/backends/cadence/aot/quantizer/
Dutils.py17 from torch.ao.quantization import ObserverOrFakeQuantize
128 obs_or_fqs: List[ObserverOrFakeQuantize], argument
/external/pytorch/test/quantization/pt2e/
Dtest_quantize_pt2e.py8 from torch.ao.quantization import observer, ObserverOrFakeQuantize, QConfigMapping
339 obs_or_fqs: List[ObserverOrFakeQuantize], argument
442 obs_or_fqs: List[ObserverOrFakeQuantize], argument