Searched refs:_PartialWrapper (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/torch/ao/quantization/fx/ |
| D | lstm_utils.py | 14 from torch.ao.quantization.observer import _PartialWrapper
23 linear_output_obs_ctr: Optional[_PartialWrapper] = None, argument
24 sigmoid_obs_ctr: Optional[_PartialWrapper] = None, argument
25 tanh_obs_ctr: Optional[_PartialWrapper] = None, argument
26 cell_state_obs_ctr: Optional[_PartialWrapper] = None, argument
27 hidden_state_obs_ctr: Optional[_PartialWrapper] = None, argument
60 def make_qconfig(obs_ctr: _PartialWrapper) -> QConfig: argument
|
| D | prepare.py | 27 from torch.ao.quantization.observer import _is_activation_post_process, _PartialWrapper
186 if isinstance(observer_or_fake_quant_ctr, _PartialWrapper):
|
| /external/pytorch/torch/ao/quantization/ |
| D | qconfig.py | 28 _PartialWrapper,
616 _PartialWrapper, Type[ObserverBase], Type[FakeQuantizeBase]
624 if isinstance(obs_or_fq1, _PartialWrapper) and isinstance(
625 obs_or_fq2, _PartialWrapper
631 def _partial_wrapper_equals(obs_or_fq1: _PartialWrapper, obs_or_fq2: _PartialWrapper): argument
|
| D | qconfig_mapping.py | 11 _PartialWrapper,
44 _FIXED_QPARAMS_OP_TO_OBSERVER: Dict[Union[Callable, str], _PartialWrapper] = {
|
| D | observer.py | 60 class _PartialWrapper: class
80 result = _PartialWrapper(p=self.p)
102 r = _PartialWrapper(partial(cls_or_self, **kwargs))
126 r = _PartialWrapper(partial(cls_or_self))
|
| /external/pytorch/torch/quantization/ |
| D | observer.py | 13 _PartialWrapper,
|