Home
last modified time | relevance | path

Searched refs:float_module (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/ns/
H A D_numeric_suite.py256 def __init__(self, q_module, float_module, logger_cls): argument
259 self.shadow_module = float_module
346 float_module: nn.Module,
372 for name, mod in float_module.named_children():
449 float_module: nn.Module,
466 float_dict = get_logger_dict(float_module)
481 float_module: nn.Module,
502 float_module.qconfig = qconfig_debug # type: ignore[assignment]
504 float_module, inplace=True, allow_list=allow_list, prepare_custom_config_dict={}
/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A Dconvert.py784 float_module = original_module
787 fused_module = float_module
788 float_module = fused_module[0] # type: ignore[index]
793 if isinstance(float_module, torch.nn.RNNCellBase):
796 weight_post_process_ih(float_module.weight_ih)
797 weight_post_process_hh(float_module.weight_hh)
806 elif isinstance(float_module, (torch.nn.LSTM, torch.nn.GRU)):
809 for wn in float_module._flat_weights_names:
810 if hasattr(float_module, wn) and wn.startswith("weight"):
811 weight = getattr(float_module, wn)
[all …]
/aosp_15_r20/external/pytorch/test/quantization/eager/
H A Dtest_quantize_eager_ptq.py880 def from_float(cls, float_module): argument
881 assert hasattr(float_module, 'qconfig')
882 observed = cls(float_module.conv)
883 observed.qconfig = float_module.qconfig
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py3158 def from_float(cls, float_module): argument
3159 assert hasattr(float_module, 'qconfig')
3160 observed = cls(float_module.linear)
3161 observed.qconfig = float_module.qconfig
3328 def from_float(cls, float_module): argument
3329 assert hasattr(float_module, 'qconfig')
3330 observed = cls(float_module.linear)
3331 observed.qconfig = float_module.qconfig
3412 def from_float(cls, float_module): argument
3413 assert hasattr(float_module, 'qconfig')
[all …]
/aosp_15_r20/external/pytorch/docs/source/
H A Dquantization.rst1003 def from_float(cls, float_module):
1004 assert hasattr(float_module, 'qconfig')
1005 observed = cls(float_module.linear)
1006 observed.qconfig = float_module.qconfig