Searched defs:reset_rng_state (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/_dynamo/ | ||
| D | testing.py | 401 def reset_rng_state(use_xla=False): function |
| /external/pytorch/test/distributed/ | ||
| D | test_dynamo_distributed.py | 52 def reset_rng_state(): function |