Searched defs:use_generator (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_vmap.py | 5293 def _reset_random(self, generator, orig_state, use_generator, seed): argument
5356 def test_factory_ops(self, device, randomness, use_generator): argument
5401 def test_randperm(self, device, randomness, use_generator): argument
5667 self, device, use_generator, randomness, batched_input argument
5732 self, device, use_generator, randomness, batched_input, batched_probability argument
5800 self, device, use_generator, randomness, batched_input, batched_other argument
5860 self, device, use_generator, randomness, batched_input argument
5918 self, device, use_generator, randomness, batched_call, batched_input argument
|