Searched defs:set_extra_state (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/checkpoint/ | ||
| H A D | test_dtensor_checkpoint.py | 67 def set_extra_state( member in MyTestModule |
| H A D | test_state_dict.py | 549 def set_extra_state(self, state): function |
| /aosp_15_r20/external/pytorch/torch/csrc/dynamo/ | ||
| H A D | extra_state.cpp | 73 void set_extra_state(PyCodeObject* code, ExtraState* extra_state) { in set_extra_state() function |
| /aosp_15_r20/external/pytorch/torch/nn/modules/ | ||
| H A D | module.py | 880 def set_extra_state(self, state: Any) -> None: member in Module |
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_nn.py | 2341 def set_extra_state(self, state): member in TestNN.test_extra_state.SubModule |