48     def test_forward(self, device, dtype, module_info, training):  argument
51                                                        requires_grad=False, training=training)
65                 m.train(training)
87     def test_factory_kwargs(self, device, dtype, module_info, training):  argument
90                                                        requires_grad=False, training=training)
103                     m.train(training)
130                         m.train(training)
139                 m.train(training)
144     def test_multiple_device_transfer(self, device, dtype, module_info, training):  argument
147 …                                                            requires_grad=False, training=training)
149                                                            requires_grad=False, training=training)
159                 m.train(training)
199     def test_repr(self, device, dtype, module_info, training):  argument
203                                                        requires_grad=False, training=training)
208             m.train(training)
215     def test_save_load(self, device, dtype, module_info, training):  argument
219                                                        requires_grad=False, training=training)
231                 m.train(training)
261     def test_check_inplace(self, device, dtype, module_info, training):  argument
266                                                        requires_grad=True, training=training)
275             m_op.train(training)
278             m_inplace.train(training)
344     def test_non_contiguous_tensors(self, device, dtype, module_info, training):  argument
349                                                        requires_grad=True, training=training)
383             m.train(training)
437     def _test_gradients_helper(self, device, dtype, module_info, training, check):  argument
441                                                        requires_grad=True, training=training)
455             m.train(training)
515     def test_grad(self, device, dtype, module_info, training):  argument
516         self._test_gradients_helper(device, dtype, module_info, training, gradcheck)
520     def test_gradgrad(self, device, dtype, module_info, training):  argument
521         self._test_gradients_helper(device, dtype, module_info, training, gradgradcheck)
528     def test_cpu_gpu_parity(self, device, dtype, module_info, training):  argument
534                 and not training
542                                                            requires_grad=True, training=training)
569             cpu_module.train(training)
571             gpu_module.train(training)
619     def test_memory_format(self, device, dtype, module_info, training):  argument
626                                                        requires_grad=True, training=training)
690                 m.train(training)
697                 if training and len(ref_diff_outputs) > 0:
752                         if training and len(diff_outputs) > 0:
784     def test_if_train_and_eval_modes_differ(self, device, dtype, module_info, training):  argument
787                                                        requires_grad=False, training=training)
789         # Run forward inputs through to see if the training flag is accessed during forward.
798             m.train(training)
800             # Remove training attribute and see if forward still works.
801             delattr(m, 'training')
808                 if "'training'" in str(e):
811 …                                 "train_and_eval_differ=False, but the training mode was found to "
820     def test_device_ctx_init(self, device, dtype, module_info, training):  argument
823                                                        requires_grad=False, training=training)
826 …                                                            requires_grad=False, training=training)
847     def test_errors(self, device, dtype, module_info, training):  argument
850                                                             requires_grad=False, training=training)
870     def test_to(self, device, dtype, module_info, training, swap, set_grad):  argument
877                                                        requires_grad=False, training=training)
959     def test_to_empty(self, device, dtype, module_info, swap, training):  argument
964                                                            requires_grad=False, training=training)