Home
last modified time | relevance | path

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

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