Lines Matching defs:forward
47 def forward(self, X): member in TestNNParametrization.test_register_and_remove_parametrization.Skew
52 def forward(self, X): member in TestNNParametrization.test_register_and_remove_parametrization.Orthogonal
63 def forward(self, X): member in TestNNParametrization.test_register_and_remove_parametrization.Resize
67 def forward(self, X): member in TestNNParametrization.test_register_and_remove_parametrization.NoResize
72 def forward(self, x): member in TestNNParametrization.test_register_and_remove_parametrization.FirstZero
76 def forward(self, x): member in TestNNParametrization.test_register_and_remove_parametrization.LastZero
299 def forward(self, X): member in TestNNParametrization.test_register_and_remove_nested_parametrization.Skew
352 def forward(self, x): member in TestNNParametrization.test_register_and_remove_buffer_parametrization.FirstZero
356 def forward(self, x): member in TestNNParametrization.test_register_and_remove_buffer_parametrization.LastZero
397 def forward(self, X): member in TestNNParametrization.test_serialization_parametrization.Orthogonal
442 def forward(self, X): member in TestNNParametrization.test_initialization_parametrization.Skew
460 def forward(self, X): member in TestNNParametrization.test_initialization_parametrization.Orthogonal
512 def forward(self, x): member in TestNNParametrization.test_errors_unparametrized_tensor_parametrization.Identity
527 def forward(self, x, y): member in TestNNParametrization.test_errors_unparametrized_tensor_parametrization.Sum
543 def forward(self, x, y, z): member in TestNNParametrization.test_errors_unparametrized_tensor_parametrization.WrongNumberParams
566 def forward(self, x, y): member in TestNNParametrization.test_errors_unparametrized_tensor_parametrization.WrongRightInverseSequence
580 def forward(self, x): member in TestNNParametrization.test_errors_unparametrized_tensor_parametrization.ChangeDtypeInverse
595 def forward(self, x): member in TestNNParametrization.test_errors_unparametrized_tensor_parametrization.NotTensor
605 def forward(self, x): member in TestNNParametrization.test_errors_unparametrized_tensor_parametrization.ChangeDtype
615 def forward(self, x): member in TestNNParametrization.test_errors_unparametrized_tensor_parametrization.ChangeShape
625 def forward(self, x, y): member in TestNNParametrization.test_errors_unparametrized_tensor_parametrization.ChangeDtypeMulti
638 def forward(self, x): member in TestNNParametrization.test_errors_unparametrized_tensor_parametrization.SequenceLen1
660 def forward(self, x): member in TestNNParametrization.test_errors_parametrized_tensor_parametrization.Identity
668 def forward(self, x): member in TestNNParametrization.test_errors_parametrized_tensor_parametrization.WrongReturn
679 def forward(self, x): member in TestNNParametrization.test_errors_parametrized_tensor_parametrization.ChangeDtype
690 def forward(self, x): member in TestNNParametrization.test_errors_parametrized_tensor_parametrization.ChangeShape
741 def forward(self, x, y): member in TestNNParametrization.test_multiple_inputs_parametrization.RankOne
754 def forward(self, x): member in TestNNParametrization.test_multiple_inputs_parametrization.Double
845 def forward(self, X): member in TestNNParametrization.test_caching_parametrization.Skew
850 def forward(self, X): member in TestNNParametrization.test_caching_parametrization.Orthogonal
872 def forward(self, X): member in TestNNParametrization.test_caching_parametrization_with_transfer_parametrizations_and_params.Skew
877 def forward(self, X): member in TestNNParametrization.test_caching_parametrization_with_transfer_parametrizations_and_params.Orthogonal
905 def forward(self, X): member in TestNNParametrization.test_parametrization_same_training_mode.Identity
922 def forward(self, X): member in TestNNParametrization.test_type_before_parametrizations.Identity
940 def forward(self, x): member in TestNNParametrization.test_deepcopy_after_parametrization.AddOne
1001 def forward(self, x): member in TestNNParametrization.test_transfer_parametrizations_and_params.AddOne
1005 def forward(self, x): member in TestNNParametrization.test_transfer_parametrizations_and_params.Double
1012 def forward(self, x): member in TestNNParametrization.test_transfer_parametrizations_and_params.MinusOne
1069 def forward(self, x): member in TestNNParametrization.test_transfer_parametrizations_and_params_right_inverse.Double
1099 def forward(self, x): member in TestNNParametrization.test_transfer_parametrizations_and_params_single_param.AddOne
1103 def forward(self, x): member in TestNNParametrization.test_transfer_parametrizations_and_params_single_param.Double
1107 def forward(self, x): member in TestNNParametrization.test_transfer_parametrizations_and_params_single_param.MinusOne
1138 def forward(self, x, y): member in TestNNParametrization.test_transfer_parametrizations_and_params_many_to_one.RankOne
1150 def forward(self, x): member in TestNNParametrization.test_transfer_parametrizations_and_params_many_to_one.Double
1424 def forward(self, x0, x1): member in TestNNParametrization.test_register_parametrization_no_grad.SplitAndCat
1745 def forward(self, X): member in TestNNParametrization.test_wrapper_subclass_parametrization.Subclassify
1749 def forward(self, X): member in TestNNParametrization.test_wrapper_subclass_parametrization.UnSubclassify
1753 def forward(self, X): member in TestNNParametrization.test_wrapper_subclass_parametrization.IdentityWithRightInverse