Lines Matching defs:forward
52 def forward(self, x): member in BasicModule
62 def forward(self, x): member in FnMember
75 def forward(self, x): member in FnMemberCmp
91 def forward(self, x): member in SubmoduleExample
104 def forward(self, x): member in IsTrainingCheck
129 def forward(self, x): member in ModuleMethodCall
146 def forward(self, x): member in UnsupportedMethodCall
157 def forward(self, x): member in UnsupportedModule
167 def forward(self, x): member in UnsupportedModuleCall
173 def forward(x): member in ModuleWithStaticForward
182 def forward(self, x): member in ModuleCallModuleWithStaticForward
198 def forward(self, x): member in ModuleStaticMethodCall
216 def forward(self, x): member in ModuleClassMethodCall
231 def forward(self, x): member in ModuleProperty
249 def forward(self, x): member in NestedModuleList
261 def forward(self, x): member in ConstLoop
272 def forward(self, x): member in ViaModuleCall
283 def forward(self, x): member in IsNoneLayer
300 def forward(self, x): member in LayerList
318 def forward(self, x): member in ModuleList
358 def forward(self, x): member in CustomGetItemModuleList
374 def forward(self, x): member in ModuleDict
389 def forward(self, x): member in ParameterDict
406 def forward(self, x): member in CustomGetItemParameterDict
423 def forward(self, x): member in CustomGetItemModuleDict
438 def forward(self, x): member in TensorList
452 def forward(self, x): member in Children
466 def forward(self, x): member in NamedChildren
477 def forward(self, x, offset=1): member in IntArg
492 def forward(self, x): member in Seq
508 def forward(self, x): member in CfgModule
520 def forward(self, x): member in StringMember
526 def forward(self, x): member in _Block
541 def forward(self, init_features): member in _DenseBlock
554 def forward(self, x): member in DenseNetBlocks
568 def forward(self, x): member in MaterializedModule
594 def forward(self, input): member in LazyMLP
615 def forward(self, input): member in LazyLayerWithNamedTupleInput
628 def forward(self, input): member in LazyModuleWithNamedTupleInput
640 def forward(self, input): member in LazyLayerWithListInput
652 def forward(self, input): member in LazyModuleWithListInput
664 def forward(self, x): member in LazyModuleWithLazySubmodule
677 def forward(self, x, y): member in LazyLayerWithInputs
695 def forward(self, x, y): member in LazyModuleKwArgs
711 def forward(self, x): member in LazyChildModuleNoClsToBecome
734 def forward(self, x): member in ParametersModule1
742 def forward(self, x): member in ParametersModule2
750 def forward(self, x): member in ParametersModule3
756 def forward(self, x): member in ParametersModule4
768 def forward(self, x): member in ParametersModule5
777 def forward(self, x): member in SuperModule
783 def forward(self, x): member in SuperModule2
800 def forward(self, x): member in SuperChildCallsClassMethod
810 def forward(self, x): member in HasAttrModule
828 def forward(self, init_features): member in EnumValues
845 def forward(self, init_features): member in AccessByKeys
859 def forward(self, x): member in CallForwardDirectly
870 def forward(self, x): member in ConvCallForwardDirectly
879 def forward(self, x): member in ConvTransposeCallForwardDirectly
892 def forward(self, inputs, mask=None): member in ConvCallSuperForwardDirectly
906 def forward(self, x): member in ConvTransposeCallSuperForwardDirectly
929 def forward(self, x): member in ModuleNameString
943 def forward(self, x): member in SelfMutatingModule
973 def forward(self, x): member in ModuleAttributePrecedence
999 def forward(self, x): member in ModuleForwardHasGraphBreak
1039 def forward(self, x): member in ModuleGuardNameIsValid
1059 def forward(self, x): member in SequentialWithDuplicatedModule
1080 def forward(self, x): member in SequentialWithDuplicatedModule2
1095 def forward(self, x): member in ModuleComparison
1110 def forward(self, x): member in ModulePatch2
1117 def forward(self, x): member in UnspecNonInlinableModule
1129 def forward(self, x): member in UnspecNonInlinableToplevelModule
1398 def forward(self, x): member in NNModuleTests.test_nn_moduledict_contains.M
1627 def forward(self, x): member in NNModuleTests.test_call_fn_with_non_const_inputs_safe.ModuleSpecialFwd
1676 def forward(self, x): member in MockModule
1698 def forward(self, x, c=4): member in OptimizedModuleTest.test_attr_precedence.Mod
1720 def forward(self, x, c=None): member in OptimizedModuleTest.test_attr_precedence.MyMod
1770 def forward(self, x): member in OptimizedModuleTest.test_param_order.MyModule
1808 def forward(self, x): member in OptimizedModuleTest.test_buffer_order.MyModule
1846 def forward(self, x): member in OptimizedModuleTest.test_module_order.MyModule
1889 def forward(self, x): member in OptimizedModuleTest.test_attr.MockModule
1919 def forward(self, x): member in OptimizedModuleTest.test_dir.MockModule
1946 def forward(self, x): member in OptimizedModuleTest.test_no_recompile_on_nn_guarded_modules.SubModule
1956 def forward(self, x): member in OptimizedModuleTest.test_no_recompile_on_nn_guarded_modules.MockModule
1985 def forward(self, x): member in OptimizedModuleTest.test_recompile_limit_on_freed_module.Mod
2012 def forward(self, x): member in OptimizedModuleTest.test_inline_inbuilt_nn_modules.SubModule
2022 def forward(self, x): member in OptimizedModuleTest.test_inline_inbuilt_nn_modules.MockModule
2050 def forward(self, x): member in OptimizedModuleTest.test_cache_size_limit_on_guarded_nn_modules.SubModule
2060 def forward(self, x): member in OptimizedModuleTest.test_cache_size_limit_on_guarded_nn_modules.MockModule
2101 def forward(self, x): member in OptimizedModuleTest.test_composition.InnerModule
2111 def forward(self, x): member in OptimizedModuleTest.test_composition.OuterModule
2129 def forward(self, x): member in OptimizedModuleTest.test_composition_with_opt_mod.InnerModule
2141 def forward(self, x): member in OptimizedModuleTest.test_composition_with_opt_mod.OuterModule
2170 def forward(self, x: torch.Tensor) -> torch.Tensor: member in OptimizedModuleTest.test_hooks_outer.TestModule
2217 def forward(self, x: torch.Tensor) -> torch.Tensor: member in OptimizedModuleTest.test_hooks_inner.TestModule
2286 def forward(self, x: torch.Tensor) -> torch.Tensor: member in OptimizedModuleTest.test_hooks_skip_guards.TestModule
2383 def forward(self, x): member in OptimizedModuleTest.test_hooks_allowed_modules.ToyModel
2398 def forward(self, x): member in OptimizedModuleTest.test_hooks_allowed_modules_compiles.ToyModel
2432 def forward(self, x): member in OptimizedModuleTest.test_hooks_allowed_modules_compiles_self_contained.ToyModel
2475 def forward(self, x): member in OptimizedModuleTest.test_dunder_call_explicitly.ToyModel
2490 def forward(self, x): member in OptimizedModuleTest.test_backward_hooks.CustomLinear
2502 def forward(self, x): member in OptimizedModuleTest.test_backward_hooks.ToyModel
2564 def forward(self, x): member in OptimizedModuleTest.test_udo_instance_method_as_hook.ToyModel
2581 def forward(self, x): member in OptimizedModuleTest.test_module_dict_iter_name.MyModule
2603 def forward(self, x): member in OptimizedModuleTest.test_module_dict_iter_keys.MyModule
2632 def forward(self, x): member in OptimizedModuleTest.test_assign_does_not_exist.MyModule
2648 def forward(self, x): member in OptimizedModuleTest.test_module_dict_iter_values.MyModule
2681 def forward(self, x): member in OptimizedModuleTest.test_no_op_assignment.Mod
2759 def forward(self, x): member in OptimizedModuleTest.test_mark_static_previously_seen_tensor.TestModule
2803 def forward(self, x): member in OptimizedModuleTest.test_mark_static_nn_module_tensor.TestModule
2841 def forward(self, x): member in OptimizedModuleTest.test_mark_static_with_freezing.TestModule
2867 def forward(self, x): member in OptimizedModuleTest.test_guard_on_torch_nn_modules.MockModule
2898 def forward(self, inp): member in OptimizedModuleTest.test_setattr_on_compiled_module.ReplayMutation
2979 def forward(self, x): member in OptimizedModuleTest.test_monkeypatching_forward.FakeModule
2986 def forward(self, x): member in OptimizedModuleTest.test_monkeypatching_forward.MyModule
3020 def forward(self, x): member in OptimizedModuleTest.test_user_defined_nn_module_dynamic.Conv2d