Searched refs:get_torch_state_as_tuple (Results 1 – 4 of 4) sorted by relevance
331 fwd_seed, fwd_base_offset = CUDARngStateHelper.get_torch_state_as_tuple(334 bwd_seed, bwd_base_offset = CUDARngStateHelper.get_torch_state_as_tuple(348 fwd_seed, fwd_base_offset = CUDARngStateHelper.get_torch_state_as_tuple(
460 seed, offset = CUDARngStateHelper.get_torch_state_as_tuple(fake_mode)478 seed, offset = CUDARngStateHelper.get_torch_state_as_tuple()1731 rng_args = CUDARngStateHelper.get_torch_state_as_tuple()
286 rng_states.append(CUDARngStateHelper.get_torch_state_as_tuple())
1977 def get_torch_state_as_tuple(fake_mode=nullcontext()): member in CUDARngStateHelper