Lines Matching defs:forward
32 def forward(self, input: Tensor) -> Tensor: member in OrigModule
40 def forward(self, input: Tensor) -> Tensor: member in NewModule
58 def forward(self, input: Tensor) -> Tensor: member in TestModuleInterface.test_not_submodule_interface_call.TestNotModuleInterfaceCall
75 def forward(self, x: Tensor) -> Tensor: member in TestModuleInterface.test_module_interface.OneTwoModule
93 def forward(self, x: Tensor) -> Tensor: member in TestModuleInterface.test_module_interface.FooMod
103 def forward(self, x: Tensor) -> Tensor: member in TestModuleInterface.test_module_interface.BarMod
159 def forward(self, input): member in TestModuleInterface.test_module_doc_string.TestInterface
173 def forward(self, input): member in TestModuleInterface.test_module_doc_string.TestModule
189 def forward(self, x: Tensor) -> Tensor: member in TestModuleInterface.test_module_interface_subtype.OneTwoModule
206 def forward(self, x: Tensor) -> Tensor: member in TestModuleInterface.test_module_interface_subtype.Foo
219 def forward(self, x: Tensor) -> Tensor: member in TestModuleInterface.test_module_interface_subtype.WrongMod
231 def forward(self, input: torch.Tensor) -> Any: member in TestModuleInterface.test_module_interface_subtype.TensorToAny
242 def forward(self, input: Any) -> Any: member in TestModuleInterface.test_module_interface_subtype.AnyToAny
252 def forward(self, input: Any) -> Any: member in TestModuleInterface.test_module_interface_subtype.TensorToAnyImplA
256 def forward(self, input: Any) -> torch.Tensor: member in TestModuleInterface.test_module_interface_subtype.TensorToAnyImplB
260 def forward(self, input: Any) -> torch.Tensor: member in TestModuleInterface.test_module_interface_subtype.AnyToAnyImpl
283 def forward(self, input: Tensor) -> Tensor: member in TestModuleInterface.test_module_swap.ModuleInterface
293 def forward(self, input: Tensor) -> Tensor: member in TestModuleInterface.test_module_swap.TestModule
316 def forward(self, input: Tensor) -> Tensor: member in TestModuleInterface.test_module_swap_wrong_module.ModuleInterface
320 def forward(self, input: int) -> int: member in TestModuleInterface.test_module_swap_wrong_module.NewModuleWrong
330 def forward(self, input: Tensor) -> Tensor: member in TestModuleInterface.test_module_swap_wrong_module.TestModule
344 def forward(self, input: Tensor) -> Tensor: member in TestModuleInterface.test_module_swap_no_lazy_compile.ModuleInterface
354 def forward(self, input: Tensor) -> Tensor: member in TestModuleInterface.test_module_swap_no_lazy_compile.TestModule
361 def forward(self, input: Tensor) -> Tensor: member in TestModuleInterface.test_module_swap_no_lazy_compile.NewModuleMethodNotLazyCompile
375 def forward(self, input: Tensor) -> Tensor: member in TestModuleInterface.test_module_swap_no_lazy_compile.NewModuleMethodManualExport
389 def forward(self, input: Tensor) -> Tensor: member in TestModuleInterface.test_module_swap_no_module_interface.TestNoModuleInterface
409 def forward(self, input: Tensor) -> Tensor: member in TestModuleInterface.test_script_module_as_interface_swap.ModuleInterface
418 def forward(self, input: Tensor) -> Tensor: member in TestModuleInterface.test_script_module_as_interface_swap.OrigScriptModule
427 def forward(self, input: Tensor) -> Tensor: member in TestModuleInterface.test_script_module_as_interface_swap.NewScriptModule
437 def forward(self, input: Tensor) -> Tensor: member in TestModuleInterface.test_script_module_as_interface_swap.TestNNModuleWithScriptModule
455 def forward(self, x): member in TestModuleInterface.test_freeze_module_with_interface.SubModule
463 def forward(self, x): member in TestModuleInterface.test_freeze_module_with_interface.OrigMod
468 def forward(self, x: Tensor) -> int: member in TestModuleInterface.test_freeze_module_with_interface.ModInterface
479 def forward(self, x): member in TestModuleInterface.test_freeze_module_with_interface.TestModule
498 def forward(self, x): member in TestModuleInterface.test_freeze_module_with_setattr_in_interface.SubModule
511 def forward(self, x): member in TestModuleInterface.test_freeze_module_with_setattr_in_interface.OrigMod
516 def forward(self, x: Tensor) -> int: member in TestModuleInterface.test_freeze_module_with_setattr_in_interface.ModInterface
527 def forward(self, x): member in TestModuleInterface.test_freeze_module_with_setattr_in_interface.TestModule
541 def forward(self, x): member in TestModuleInterface.test_freeze_module_with_inplace_mutation_in_interface.SubModule
554 def forward(self, x): member in TestModuleInterface.test_freeze_module_with_inplace_mutation_in_interface.OrigMod
559 def forward(self, x: Tensor) -> Tensor: member in TestModuleInterface.test_freeze_module_with_inplace_mutation_in_interface.ModInterface
570 def forward(self, x): member in TestModuleInterface.test_freeze_module_with_inplace_mutation_in_interface.TestModule
587 def forward(self, x): member in TestModuleInterface.test_freeze_module_with_mutated_interface.SubModule
599 def forward(self, x): member in TestModuleInterface.test_freeze_module_with_mutated_interface.OrigMod
604 def forward(self, x: Tensor) -> Tensor: member in TestModuleInterface.test_freeze_module_with_mutated_interface.ModInterface
615 def forward(self, x): member in TestModuleInterface.test_freeze_module_with_mutated_interface.TestModule
634 def forward(self, x): member in TestModuleInterface.test_freeze_module_with_interface_and_fork.SubModule
643 def forward(self, x): member in TestModuleInterface.test_freeze_module_with_interface_and_fork.OrigMod
648 def forward(self, x: Tensor) -> Tensor: member in TestModuleInterface.test_freeze_module_with_interface_and_fork.ModInterface
659 def forward(self, x): member in TestModuleInterface.test_freeze_module_with_interface_and_fork.TestModule
669 def forward(self, x): member in TestModuleInterface.test_freeze_module_with_interface_and_fork.MainModule
692 def forward(self, input): member in TestModuleInterface.test_module_apis_interface.TestModule