Searched defs:get_rng_state (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | random.py | 22 def get_rng_state() -> torch.Tensor: function
|
| /aosp_15_r20/external/pytorch/torch/cuda/ |
| H A D | random.py | 23 def get_rng_state(device: Union[int, str, torch.device] = "cuda") -> Tensor: function
|
| /aosp_15_r20/external/pytorch/torch/xpu/ |
| H A D | random.py | 10 def get_rng_state(device: Union[int, str, torch.device] = "xpu") -> Tensor: function
|
| /aosp_15_r20/external/pytorch/torch/mps/ |
| H A D | __init__.py | 36 def get_rng_state(device: Union[int, str, torch.device] = "mps") -> Tensor: function
|
| /aosp_15_r20/external/pytorch/torch/mtia/ |
| H A D | __init__.py | 284 def get_rng_state(device: Union[int, str, torch.device] = "mtia") -> Tensor: function
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_cpp_extensions_open_device_registration.py | 46 def get_rng_state(device: Union[int, str, torch.device] = "foo") -> torch.Tensor: function
|