Searched defs:get_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 | 61 def get_extra_state(self) -> Dict[str, Union[int, torch._tensor.Tensor]]: member in MyTestModule |
| H A D | test_state_dict.py | 546 def get_extra_state(self): function |
| /aosp_15_r20/external/pytorch/torch/csrc/dynamo/ | ||
| H A D | extra_state.cpp | 60 ExtraState* get_extra_state(PyCodeObject* code) { in get_extra_state() function |
| /aosp_15_r20/external/pytorch/torch/nn/modules/ | ||
| H A D | module.py | 859 def get_extra_state(self) -> Any: member in Module |
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_nn.py | 2336 def get_extra_state(self): member in TestNN.test_extra_state.SubModule |