Lines Matching defs:forward
58 def forward(self, x): member in TestUnflatten.test_unflatten_nested.NestedChild
69 def forward(self, x): member in TestUnflatten.test_unflatten_nested.Child1
78 def forward(self, x): member in TestUnflatten.test_unflatten_nested.Child2
90 def forward(self, x): member in TestUnflatten.test_unflatten_nested.MyModule
120 def forward(self, x): member in TestUnflatten.test_unflatten_buffer_mutation.Child
132 def forward(self, x): member in TestUnflatten.test_unflatten_buffer_mutation.MyModule
156 def forward(self, x): member in TestUnflatten.test_unflatten_nested_access.Child
167 def forward(self, x): member in TestUnflatten.test_unflatten_nested_access.MyModule
191 def forward(self, x): member in TestUnflatten.test_unflatten_shared_submodule.Shared
211 def forward(self, zx, y): member in TestUnflatten.test_unflatten_preserve_signature.NestedChild
219 def forward(self, x, y): member in TestUnflatten.test_unflatten_preserve_signature.Child1
228 def forward(self, x): member in TestUnflatten.test_unflatten_preserve_signature.Child2
237 def forward(self, x, y): member in TestUnflatten.test_unflatten_preserve_signature.MyModule
294 def forward(self, x): member in TestUnflatten.test_unflatten_param_list_dict.Mod
310 def forward(self, x, a, b): member in TestUnflatten.test_unflatten_preserve_with_unused_input.M1
318 def forward(self, x, y): member in TestUnflatten.test_unflatten_preserve_with_unused_input.M
344 def forward(self, x): member in TestUnflatten.test_unflatten_wrong_input.Mod
368 def forward(self, x): member in TestUnflatten.test_unflatten_with_inplace_compile.NestedChild
379 def forward(self, x): member in TestUnflatten.test_unflatten_with_inplace_compile.Child1
388 def forward(self, x): member in TestUnflatten.test_unflatten_with_inplace_compile.Child2
400 def forward(self, x): member in TestUnflatten.test_unflatten_with_inplace_compile.MyModule
424 def forward(self, x, y): member in TestUnflatten.test_fx_trace.MyModule
443 def forward(self, x): member in TestUnflatten.test_double_nested_submodule.SubSubMod
451 def forward(self, x): member in TestUnflatten.test_double_nested_submodule.SubMod
459 def forward(self, x): member in TestUnflatten.test_double_nested_submodule.MyModule
475 def forward(self, x): member in TestUnflatten.test_unflatten_container_type.Leaf
484 def forward(self, x, z): member in TestUnflatten.test_unflatten_container_type.Bar
492 def forward(self, x, z): member in TestUnflatten.test_unflatten_container_type.Foo
510 def forward(self, x): member in TestUnflatten.test_unflattened_module_nodes_has_meta_val.SubMod
518 def forward(self, x): member in TestUnflatten.test_unflattened_module_nodes_has_meta_val.MyModule
543 def forward(self, x): member in TestUnflatten.test_unflatten_requires_grad_param.M
561 def forward(self, x): member in TestUnflatten.test_placeholder_and_get_attr_ordering_after_unflattened.TransposeModule
587 def forward(self, x): member in TestUnflatten.test_unflatten_constant_tensor.SubMod
595 def forward(self, x): member in TestUnflatten.test_unflatten_constant_tensor.Mod
627 def forward(self, x): member in TestUnflatten.test_unflatten_constant_obj.SubMod
635 def forward(self, x): member in TestUnflatten.test_unflatten_constant_obj.Mod
655 def forward(self, x): member in TestUnflatten.test_unflatten_skipped_call_module.C
663 def forward(self, x): member in TestUnflatten.test_unflatten_skipped_call_module.B
670 def forward(self, x): member in TestUnflatten.test_unflatten_skipped_call_module.D
679 def forward(self, x): member in TestUnflatten.test_unflatten_skipped_call_module.A
691 def forward(self, x): member in TestUnflatten.test_nested_leaf_non_strict.Leaf
699 def forward(self, x): member in TestUnflatten.test_nested_leaf_non_strict.Nested
707 def forward(self, x): member in TestUnflatten.test_nested_leaf_non_strict.TopLevel
726 def forward(self, x): member in TestUnflatten.test_unflatten_submodule_ordering.Module2
735 def forward(self, x): member in TestUnflatten.test_unflatten_submodule_ordering.Module1
745 def forward(self, x): member in TestUnflatten.test_unflatten_submodule_ordering.Module
775 def forward(self, input_): member in TestUnflatten.test_duplicate_placeholder.MyModule
802 def forward(self, x): member in TestUnflatten.test_simple_alias.Foo
821 def forward(self, x): member in TestUnflatten.test_simple_alias.Foo
832 def forward(self, x, const) -> torch.Tensor: member in TestUnflatten.test_attr_as_submod_input.layer
841 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestUnflatten.test_attr_as_submod_input.M
859 def forward(self, x, y): member in TestUnflatten.test_dedup_sym_size.M1
864 def forward(self, x, y): member in TestUnflatten.test_dedup_sym_size.M2
874 def forward(self, x, y): member in TestUnflatten.test_dedup_sym_size.M
902 def forward(self, x): member in TestUnflatten.test_unflatten_eager.NestedChild
913 def forward(self, x): member in TestUnflatten.test_unflatten_eager.Child1
922 def forward(self, x): member in TestUnflatten.test_unflatten_eager.Child2
934 def forward(self, x): member in TestUnflatten.test_unflatten_eager.MyModule