Lines Matching defs:forward
77 def forward(self, x): member in TestFreezing.test_freeze_module.M
135 def forward(self, x): member in TestFreezing.test_freeze_module_with_submodule.SubModule
144 def forward(self, x): member in TestFreezing.test_freeze_module_with_submodule.SubModule2
156 def forward(self, x): member in TestFreezing.test_freeze_module_with_submodule.TestModule
200 def forward(self, x): member in TestFreezing.test_freeze_module_with_fork.SubModule
208 def forward(self, x): member in TestFreezing.test_freeze_module_with_fork.TestModule
240 def forward(self, x): member in TestFreezing.test_freeze_module_with_nested_fork.SubModule
249 def forward(self, x): member in TestFreezing.test_freeze_module_with_nested_fork.SubModule2
261 def forward(self, x): member in TestFreezing.test_freeze_module_with_nested_fork.TestModule
299 def forward(self, x): member in TestFreezing.test_freeze_module_with_fork2.TestModule
348 def forward(self, x): member in TestFreezing.test_freeze_module_with_fork_calling_module_method.TestModule
380 def forward(self, x): member in TestFreezing.test_freeze_module_with_sharedclasstype.SubModule
399 def forward(self, x): member in TestFreezing.test_freeze_module_with_sharedclasstype.SubModule2
410 def forward(self, x): member in TestFreezing.test_freeze_module_with_sharedclasstype.TestModule
472 def forward(self, x): member in TestFreezing.test_freeze_module_with_nestedaliasing.SubModule
492 def forward(self, x): member in TestFreezing.test_freeze_module_with_nestedaliasing.SubModule2
501 def forward(self, x): member in TestFreezing.test_freeze_module_with_nestedaliasing.TestModule
531 def forward(self, x): member in TestFreezing.test_freeze_module_with_nestedaliasingscalar.SubModule
551 def forward(self, x): member in TestFreezing.test_freeze_module_with_nestedaliasingscalar.SubModule2
560 def forward(self, x): member in TestFreezing.test_freeze_module_with_nestedaliasingscalar.TestModule
588 def forward(self, x): member in TestFreezing.test_freeze_module_with_preserve_sub_module.SubModule
597 def forward(self, x): member in TestFreezing.test_freeze_module_with_preserve_sub_module.TestModule
622 def forward(self, x): member in TestFreezing.test_freeze_module_with_preserve_sub_module_and_mutation.SubModule
632 def forward(self, x): member in TestFreezing.test_freeze_module_with_preserve_sub_module_and_mutation.TestModule
660 def forward(self, x): member in TestFreezing.test_freeze_module_with_helperfunction.SubModule
670 def forward(self, x): member in TestFreezing.test_freeze_module_with_helperfunction.TestModule
696 def forward(self, x): member in TestFreezing.test_freeze_module_with_inplace_mutable.FreezeMe
717 def forward(self, x): member in TestFreezing.test_freeze_module_with_mutable_list.FreezeMe
744 def forward(self, x): member in TestFreezing.test_freeze_module_with_mutable_dict.FreezeMe
773 def forward(self, x): member in TestFreezing.test_freeze_module_with_mutable_tensor.FreezeMe
795 def forward(self, x): member in TestFreezing.test_freeze_module_with_tuple.FreezeMe
817 def forward(self, x): member in TestFreezing.test_freeze_module_with_tensor.FreezeMe
839 def forward(self, x): member in TestFreezing.test_freeze_module_with_list.FreezeMe
861 def forward(self, x): member in TestFreezing.test_freeze_module_with_aliased_tensor_attr.FreezeMe
884 def forward(self, x): member in TestFreezing.test_freeze_module_with_aliased_tensor_attr2.FreezeMe
905 def forward(self, x): member in TestFreezing.test_freeze_module_with_aliased_tensor_attr3.FreezeMe
928 def forward(self, x): member in TestFreezing.test_freeze_module_with_aliased_tensor_attr4.FreezeMe
952 def forward(self, x): member in TestFreezing.test_freeze_module_with_overlapping_attrs.FreezeMe
975 def forward(self, x): member in TestFreezing.test_freeze_module_with_aliased_attr.FreezeMe
1001 def forward(self, x): member in TestFreezing.test_freeze_module_with_aliased_attr2.FreezeMe
1025 def forward(self, x): member in TestFreezing.test_freeze_module_with_aliased_attr3.FreezeMe
1048 def forward(self, x): member in TestFreezing.test_freeze_module_return_self.FreezeMe
1071 def forward(self, i: int): member in TestFreezing.test_freeze_module_inlining.Mod
1093 def forward(self, x): member in TestFreezing.test_freeze_module_return_sub_module.FreezeMe
1147 def forward(self, x): member in TestFreezing.test_freeze_module_in_training_mode.Net
1288 def forward(self, x): member in TestFreezing.test_freeze_module_with_user_preserved_attr.Module
1305 def forward(self, x): member in TestFreezing.test_freeze_module_with_user_preserved_method.Module
1336 def forward(self, x): member in TestFreezing.test_freeze_module_with_user_preserved_method2.Module
1358 def forward(self): member in TestFreezing.test_freeze_module_with_user_preserved_attribute_on_submodule.SubModule
1367 def forward(self): member in TestFreezing.test_freeze_module_with_user_preserved_attribute_on_submodule.Module
1392 def forward(self): member in TestFreezing.test_freeze_module_with_user_preserved_attribute_on_unused_submodule.SubModule
1404 def forward(self): member in TestFreezing.test_freeze_module_with_user_preserved_attribute_on_unused_submodule.Module
1418 def forward(self, x): member in TestFreezing.test_freeze_module_with_user_preserved_method_on_submodule.SubModule
1432 def forward(self, x): member in TestFreezing.test_freeze_module_with_user_preserved_method_on_submodule.Module
1450 def forward(self, x): member in TestFreezing.test_module_with_shared_type_instances.Child
1463 def forward(self, x): member in TestFreezing.test_module_with_shared_type_instances.Parent
1509 def forward(self, cond: bool): member in TestFreezing.test_module_getattr_indirection.Mod
1531 def forward(self, inp: Any) -> Any: member in TestFreezing.test_freeze_module_with_non_static_module_container_index.ModuleInterface
1535 def forward(self, inp: Any) -> Any: member in TestFreezing.test_freeze_module_with_non_static_module_container_index.ImplementsInterface
1546 def forward(self, x: torch.Tensor, key: str) -> Any: member in TestFreezing.test_freeze_module_with_non_static_module_container_index.ModWithDict
1563 def forward(self, x: torch.Tensor, idx: int) -> Any: member in TestFreezing.test_freeze_module_with_non_static_module_container_index.ModWithList
1578 def forward(self, inp: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_with_interface_mutable.ModuleInterface
1586 def forward(self, inp: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_with_interface_mutable.ImplementsInterface
1597 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_with_interface_mutable.WrapperModule
1612 def forward(self, inp: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_with_swapping_interfaces.ModuleInterface
1616 def forward(self, inp: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_with_swapping_interfaces.Implementation1
1620 def forward(self, inp: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_with_swapping_interfaces.Implementation2
1633 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_with_swapping_interfaces.WrapperModule
1651 def forward(self, inp: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_recursive_interfaces.InnerInterface
1656 def forward(self, inp: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_recursive_interfaces.OuterInterface
1664 def forward(self, inp): member in TestFreezing.test_freeze_recursive_interfaces.InnerImpl
1674 def forward(self, inp): member in TestFreezing.test_freeze_recursive_interfaces.OuterImpl
1684 def forward(self, inp): member in TestFreezing.test_freeze_recursive_interfaces.WrapperModule
1701 def forward(self, inp: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_recursive_interfaces_with_reassignment.InnerInterface
1706 def forward(self, inp: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_recursive_interfaces_with_reassignment.OuterInterface
1714 def forward(self, inp): member in TestFreezing.test_freeze_recursive_interfaces_with_reassignment.InnerImpl1
1722 def forward(self, inp): member in TestFreezing.test_freeze_recursive_interfaces_with_reassignment.InnerImpl2
1735 def forward(self, inp): member in TestFreezing.test_freeze_recursive_interfaces_with_reassignment.OuterImpl
1750 def forward(self, inp): member in TestFreezing.test_freeze_recursive_interfaces_with_reassignment.WrapperModule
1765 def forward(self, inp: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_interface_swapping_two_methods.MyInterface
1769 def forward(self, inp): member in TestFreezing.test_freeze_interface_swapping_two_methods.Impl1
1773 def forward(self, inp): member in TestFreezing.test_freeze_interface_swapping_two_methods.Impl2
1786 def forward(self, x): member in TestFreezing.test_freeze_interface_swapping_two_methods.WrapperModule1
1808 def forward(self, x): member in TestFreezing.test_freeze_interface_swapping_two_methods.WrapperModule2
1839 def forward(self, inp: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_recursive_interfaces_same_name.InnerInterface
1844 def forward(self, inp: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_recursive_interfaces_same_name.OuterInterface
1852 def forward(self, inp): member in TestFreezing.test_freeze_recursive_interfaces_same_name.InnerImpl
1863 def forward(self, inp): member in TestFreezing.test_freeze_recursive_interfaces_same_name.OuterImpl
1876 def forward(self, inp): member in TestFreezing.test_freeze_recursive_interfaces_same_name.WrapperModule
1896 def forward(self, inp: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_non_interface_module_swap.InnerModule
1907 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_non_interface_module_swap.WrapperModule
1933 def forward(self, inp: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_interface_within_object.MyIface
1937 def forward(self, inp: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_interface_within_object.MyImpl
1954 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestFreezing.test_freeze_interface_within_object.WrapperModule
1985 def forward(self, input): member in TestFreezing.test_freeze_non_module_class_getattr.MyModule
1998 def forward(self, x): member in TestFreezing.test_freeze_module_with_tupleoutput_submodule.SubModule
2006 def forward(self, x): member in TestFreezing.test_freeze_module_with_tupleoutput_submodule.TestModule
2028 def forward(self, x): member in TestFreezing.test_freeze_module_with_call_method.Mod
2085 def forward(self, x): member in TestFrozenOptimizations.test_conv_bn_folding.ConvBN
2137 def forward(self, x): member in TestFrozenOptimizations.test_conv_bn_folding_not_forward.ConvBN
2178 def forward(self, x): member in TestFrozenOptimizations.test_conv_bn_folding_autocast_scenario_cuda.ConvBN
2222 def forward(self, x): member in TestFrozenOptimizations.test_conv_add_folding.test_conv_fusion.ConvOp
2327 def forward(self, x): member in TestFrozenOptimizations.test_conv_mul_add_bn.Conv_Mul_Add_Bn
2367 def forward(self, x): member in TestFrozenOptimizations.test_linear_bn_folding.LinearBN
2505 def forward(self, x): member in TestFrozenOptimizations.test_linear_bn_folding_autocast_scenario_cuda.LinearBN
2555 def forward(self, in_tensor1): member in TestFrozenOptimizations.test_linear_concat.ModMultLinear
2582 def forward(self, in_tensor1): member in TestFrozenOptimizations.test_linear_concat_complex.ModMultLinear
2609 def forward(self, in_tensor1, in_tensor2): member in TestFrozenOptimizations.test_linear_concat_different_input.ModMultLinear
2629 def forward(self, in_tensor1, in_tensor2, cond: bool): member in TestFrozenOptimizations.test_linear_multiple_blocks.ModMultLinear
2711 def forward(self, x): member in TestFrozenOptimizations.test_freeze_remove_dropout.Net
2732 def forward(self, x): member in TestFrozenOptimizations.test_freeze_remove_feature_dropout.Net
2800 def forward(self, x): member in TestFrozenOptimizations.test_linear_transpose.ModLinear
2815 def forward(self, x, weight): member in TestFrozenOptimizations.test_linear_non_constant_weight.ModLinear
2887 def forward(self, x): member in TestFrozenOptimizations.test_mkldnn_fuser_broadcasting.Add
2920 def forward(self, x): member in TestFrozenOptimizations.test_mkldnn_inplace_removal.AddMul
2994 def forward(self, x): member in TestFrozenOptimizations.test_freeze_conv_relu_fusion.Net
3048 def forward(self, x): member in TestFrozenOptimizations.test_freeze_conv_relu_fusion_not_forward.Net
3102 def forward(self, i0): member in TestFrozenOptimizations.test_numel_less_than_size_with_padding.MyModule
3130 def forward(self, x): member in TestFrozenOptimizations.test_incompatible_perf_formats.Mod
3240 def forward(self, x): member in TestFrozenOptimizations.test_conv_hardswish.Clamp
3321 def forward(self, x): member in TestFrozenOptimizations.test_scalar_mul.Mod
3335 def forward(self, x): member in TestFrozenOptimizations.test_remove_detach.Mod
3347 def forward(self, x): member in TestFrozenOptimizations.test_remove_detach_not_applied.Mod
3401 def forward(self, x): member in TestMKLDNNReinplacing.test_merge_liveness.Mod
3419 def forward(self, x): member in TestMKLDNNReinplacing.test_always_alive_values.Mod
3438 def forward(self, x): member in TestMKLDNNReinplacing.test_always_alive_values.Mod
3454 def forward(self, x): member in TestMKLDNNReinplacing.test_switch_inputs_to_inplace.Mod