Searched defs:set_rng_state (Results 1 – 6 of 6) sorted by relevance
10 def set_rng_state(new_state: torch.Tensor) -> None: function
53 def set_rng_state( function
40 def set_rng_state( function
46 def set_rng_state( function
299 def set_rng_state( function
50 def set_rng_state( function