Searched defs:_get_rng_state_offset (Results 1 – 2 of 2) sorted by relevance
422 def _get_rng_state_offset(device: Union[int, str, torch.device] = "xpu") -> int: function
1276 def _get_rng_state_offset(device: Union[int, str, torch.device] = "cuda") -> int: function