Lines Matching defs:training
48 def test_forward(self, device, dtype, module_info, training): argument
87 def test_factory_kwargs(self, device, dtype, module_info, training): argument
144 def test_multiple_device_transfer(self, device, dtype, module_info, training): argument
199 def test_repr(self, device, dtype, module_info, training): argument
215 def test_save_load(self, device, dtype, module_info, training): argument
261 def test_check_inplace(self, device, dtype, module_info, training): argument
344 def test_non_contiguous_tensors(self, device, dtype, module_info, training): argument
437 def _test_gradients_helper(self, device, dtype, module_info, training, check): argument
515 def test_grad(self, device, dtype, module_info, training): argument
520 def test_gradgrad(self, device, dtype, module_info, training): argument
528 def test_cpu_gpu_parity(self, device, dtype, module_info, training): argument
619 def test_memory_format(self, device, dtype, module_info, training): argument
784 def test_if_train_and_eval_modes_differ(self, device, dtype, module_info, training): argument
820 def test_device_ctx_init(self, device, dtype, module_info, training): argument
847 def test_errors(self, device, dtype, module_info, training): argument
870 def test_to(self, device, dtype, module_info, training, swap, set_grad): argument
959 def test_to_empty(self, device, dtype, module_info, swap, training): argument