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.py | 256 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 D | convert.py | 784 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 D | test_quantize_eager_ptq.py | 880 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 D | test_quantize_fx.py | 3158 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 D | quantization.rst | 1003 def from_float(cls, float_module): 1004 assert hasattr(float_module, 'qconfig') 1005 observed = cls(float_module.linear) 1006 observed.qconfig = float_module.qconfig
|