| H A D | test_vmap.py | 5300 def _get_image(self, batched_input, batch_size, device): argument
5450 def test_dropout(self, device, randomness, batched_input): argument
5485 def test_alpha_dropout(self, device, randomness, batched_input): argument
5516 def test_feature_dropout(self, device, randomness, batched_input, dim): argument
5564 def test_feature_alpha_dropout(self, device, randomness, batched_input): argument
5608 def test_like_functions(self, device, randomness, batched_input): 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
[all …]
|