Home
last modified time | relevance | path

Searched defs:keep_vars (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/
H A Dlinear.py72 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 Dembedding_ops.py62 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
H A Dfunctional_modules.py199 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
H A Dconv.py159 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
/aosp_15_r20/external/pytorch/torch/ao/nn/sparse/quantized/
H A Dlinear.py56 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 Dtest_load_state_dict.py231 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 Dtest_module_hooks.py790 def my_state_dict_pre_hook(module, prefix, keep_vars): argument
/aosp_15_r20/external/pytorch/torch/ao/nn/sparse/quantized/dynamic/
H A Dlinear.py71 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/
H A Drnn.py65 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 Dgraph_module.py133 def _save_packed_weight(self, destination, prefix, keep_vars): argument
H A D_lower_to_native_backend.py415 def _save_packed_weight(self, destination, prefix, keep_vars): argument
/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dlazy.py193 def _save_to_state_dict(self: _LazyProtocol, destination, prefix, keep_vars): argument
H A Dmodule.py2101 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 Dutils.py149 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dfake_quantize.py266 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
H A Dobserver.py1331 def _save_to_state_dict(self, destination, prefix, keep_vars): argument
/aosp_15_r20/external/pytorch/torch/jit/
H A D_trace.py68 def _unique_state_dict(module, keep_vars=False): argument