/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/ |
H A D | linear.py | 72 def _save_to_state_dict(self, destination, prefix, keep_vars): argument 223 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
|
H A D | embedding_ops.py | 62 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
|
H A D | functional_modules.py | 199 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
|
H A D | conv.py | 159 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
|
/aosp_15_r20/external/pytorch/torch/ao/nn/sparse/quantized/ |
H A D | linear.py | 56 def _save_to_state_dict(self, destination, prefix, keep_vars): argument 164 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
|
/aosp_15_r20/external/pytorch/test/nn/ |
H A D | test_load_state_dict.py | 231 def _save_to_state_dict(self, destination, prefix, keep_vars): argument 265 def test_load_state_dict_assign_meta(self, keep_vars): argument
|
H A D | test_module_hooks.py | 790 def my_state_dict_pre_hook(module, prefix, keep_vars): argument
|
/aosp_15_r20/external/pytorch/torch/ao/nn/sparse/quantized/dynamic/ |
H A D | linear.py | 71 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
|
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/ |
H A D | rnn.py | 65 def _save_to_state_dict(self, destination, prefix, keep_vars): argument 1168 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
|
/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/ |
H A D | graph_module.py | 133 def _save_packed_weight(self, destination, prefix, keep_vars): argument
|
H A D | _lower_to_native_backend.py | 415 def _save_packed_weight(self, destination, prefix, keep_vars): argument
|
/aosp_15_r20/external/pytorch/torch/nn/modules/ |
H A D | lazy.py | 193 def _save_to_state_dict(self: _LazyProtocol, destination, prefix, keep_vars): argument
|
H A D | module.py | 2101 def _save_to_state_dict(self, destination, prefix, keep_vars): argument 2145 def state_dict(self, *args, destination=None, prefix="", keep_vars=False): argument
|
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/reference/modules/ |
H A D | utils.py | 149 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
|
/aosp_15_r20/external/pytorch/torch/ao/quantization/ |
H A D | fake_quantize.py | 266 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
|
H A D | observer.py | 1331 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
|
/aosp_15_r20/external/pytorch/torch/jit/ |
H A D | _trace.py | 68 def _unique_state_dict(module, keep_vars=False): argument
|