Lines Matching defs:forward
75 def forward(self, x): member in SimpleTest
159 def forward(self, x): member in Add
200 def forward(self, x): member in TestFX.test_graph_module.MySub
210 def forward(self, A, B, c): member in TestFX.test_graph_module.MyModule
220 def forward(self, A): member in TestFX.test_graph_module.M2
229 def forward(self, A, b=4, *args, c=5, **kwargs): member in TestFX.test_graph_module.T
238 def forward(self, x): member in TestFX.test_graph_module.M3
251 def forward(self, a): member in TestFX.test_informative_co_filename.MyModule
270 def forward(self, *args, **kwargs): member in TestFX.test_args_kwargs.T
279 def forward(self, *args, **kwargs): member in TestFX.test_varargs_concrete.T
294 def forward(*args, **kwargs): # noqa: B902 member in TestFX.test_args_kwargs_no_self.T
304 def forward(self, x): member in TestFX.test_fx_shifts.MyModule
314 def forward(self, x): member in TestFX.test_fx_and_or.MyModule
324 def forward(self, d): member in TestFX.test_dict.MyDictMod
376 def forward(self, x): member in TestFX.test_disallow_override.MyInplaceMod
387 def forward(self, x): member in TestFX.test_disallow_override.MyInplaceMod2
396 def forward(self, x): member in TestFX.test_disallow_override.MyInplaceMod3
416 def forward(self, x): member in TestFX.test_leaf_module.MyReluMod
484 def forward(self, x: torch.Tensor): member in TestFX.test_wrap_with_submodule.M
517 def forward(self, a, b): member in TestFX.test_graph_edit_with_proxy.M
536 def forward(self, x): member in TestFX.test_proxy_deepcopy_without_tracer.MyModule
563 def forward(self, x): member in TestFX.test_proxy_deepcopy_with_tracer.MyModule
583 def forward(self, x, val=None): member in TestFX.test_concrete_arg_none_assert.Foo
596 def forward(self, x, y): member in TestFX.test_trace_multiple_funcs.Foo
631 def forward(self, a, b): member in TestFX.test_graph_unique_names.M
649 def forward(self, a, b): member in TestFX.test_stack_traces.M
671 def forward(self, a, b): member in TestFX.test_stack_traces_with_transformer.M
690 def forward(self, a, b): member in TestFX.test_lineno_map.M
726 def forward(self, a, b): member in TestFX.test_unpack.M
757 def forward(self, x): member in TestFX.test_native_callable.MySimpleMod
888 def forward(self, a): member in TestFX.test_reserved_getattr.M
909 def forward(self, x): member in TestFX.test_trace_buffer_slice.ExampleCode
938 def forward(self, a, b): member in TestFX.test_node_tagging.M
954 def forward(self, x): member in TestFX.test_tensor_attribute.TensorAttribute
966 def forward(self, x): member in TestFX.test_tensor_attribute.WrapperForQualname
1003 def forward(self, x): member in TestFX.test_symbolic_trace_sequential.Simple
1018 def forward(self, x): member in TestFX.test_tensor_constant.ConstTensor
1032 def forward(self, x): member in TestFX.test_pickle_graphmodule.Nested
1099 def forward(self, x): member in TestFX.test_deepcopy_with_submods_params.Bar
1108 def forward(self, x): member in TestFX.test_deepcopy_with_submods_params.Baz
1132 def forward(self, a, b): member in TestFX.test_unpack_list_better_error.SomeArgs
1140 def forward(self, x : list): member in TestFX.test_unpack_list_better_error.UnpacksList
1149 def forward(self, x=3, y=4): member in TestFX.test_unpack_dict_better_error.SomeKwargs
1157 def forward(self, x : dict): member in TestFX.test_unpack_dict_better_error.UnpacksDict
1169 def forward(self, x): member in TestFX.test_pretty_print_targets.SomeMod
1186 def forward(self, x: torch.Tensor, y: int = 2): member in TestFX.test_pretty_print_node.M
1208 def forward(self, x): member in TestFX.test_script_tensor_constant.IHaveATensorConstant
1216 def forward(self, x): member in TestFX.test_autowrap_functions.AutowrapFnTest
1220 def forward(self, x): member in TestFX.test_autowrap_functions.AutowrapFnTest2
1257 def forward(self, x): member in TestFX.test_torch_fx_len.FXLenTest
1276 def forward(self, x): member in TestFX.test_torch_fx_len.FXLenTest2
1286 def forward(self, x): member in TestFX.test_torch_fx_getattr.FXGetattrTest
1294 def forward(self, x): member in TestFX.test_sqrt.Sqrt1
1298 def forward(self, x): member in TestFX.test_sqrt.Sqrt2
1302 def forward(self, x): member in TestFX.test_sqrt.Sqrt3
1313 def forward(self, a): member in TestFX.test_torch_custom_ops.M
1327 def forward(self, a): member in TestFX.test_torch_op_overloads.M
1345 def forward(self, a): member in TestFX.test_pickle_torch_custom_ops.M
1368 def forward(self, x): member in TestFX.test_pretty_print_graph.KwargPrintTest
1552 def forward(self, x: torch.Tensor) -> Tuple[torch.Tensor, torch.Tensor]: member in TestFX.test_return_tuple.M
1583 def forward(self, x): member in TestFX.test_symbolic_trace_assert.AssertsTensorShape
1617 def forward(self, o: CustomArgObject): member in TestFX.test_fx_create_arg.HasCustomArgObjectWhenLeaf
1629 def forward(self, x, y): member in TestFX.test_fx_create_arg.Root
1687 def forward(self, x): member in TestFX.test_example_shape_prop.TestCase
1716 def forward(self, x): member in TestFX.test_shape_prop_layout.ConvTest
1740 def forward(self, x): member in TestFX.test_shape_prop_aggregate.ReturnTwo
1748 def forward(self, x): member in TestFX.test_shape_prop_aggregate.UnderTest
1775 def forward(self, x): member in TestFX.test_shape_prop_layout_3d.ConvTest3d
1801 def forward(self, x): member in TestFX.test_nn_module_stack.SubModule
1809 def forward(self, x): member in TestFX.test_nn_module_stack.MyModule
1831 def forward(self, x): member in TestFX.test_transformer_preserves_nn_module_stack_for_get_attr.M
1856 def forward(self, x): member in TestFX.test_interpreter.MyModule
1874 def forward(self, x): member in TestFX.test_interpreter_other_graph.MyModule
1892 def forward(self, x): member in TestFX.test_interpreter_run_node_override.MyModule
1942 def forward(self, x): member in TestFX.test_interpreter_partial_eval.MyModule
1984 def forward(self, x, y=3.14159): member in TestFX.test_interpreter_default_args.Model
1997 def forward(self, x, y): member in TestFX.test_interpreter_not_enough_args.Model
2016 def forward(self, x): member in TestFX.test_transformer_noop.MyModule
2057 def forward(self, x): member in TestFX.test_transformer_multi_outputs.MyModule
2072 def forward(self, p : Pair, z : torch.Tensor, i : int) -> Dict[str, torch.Tensor]: member in TestFX.test_fn_type_annotations.Foo
2083 def forward(a : List[torch.Tensor]): function
2096 def forward(self, x, w): member in TestFX.test_wrapped_method.Foo
2121 def forward(self, x): member in TestFX.test_ctx_mgr.M
2138 def forward(self, x): member in TestFX.test_layout.M
2147 def forward(self, x, y): member in TestFX.test_ellipsis.M
2156 def forward(self, x): member in TestFX.test_inf_nan.FooMod
2371 def forward(self, x): member in TestFX.test_inline_graph.InlineInto
2375 def forward(self, x): member in TestFX.test_inline_graph.ToInline
2447 def forward(self, d : torch.Tensor): member in TestFX.test_trace_return_dataclass.ModuleReturnDataclass
2471 def forward(self, d : torch.Tensor): member in TestFX.test_trace_return_dataclass_nested.ModuleReturnDataclass
2479 def forward(self, x): member in TestFX.test_trace_return_dataclass_nested.CallsModule
2501 def forward(self, d : torch.Tensor): member in TestFX.test_trace_return_namedtuple.ModuleReturnNamedTuple
2516 def forward(self, d : Dict[int, torch.Tensor]): member in TestFX.test_trace_dict_int_keys.ModWithDictArg
2524 def forward(self, x): member in TestFX.test_trace_dict_int_keys.CallsModWithDict
2535 def forward(self, d : Dict[torch.Tensor, torch.Tensor]): member in TestFX.test_trace_dict_proxy_keys.ModWithDictArg
2543 def forward(self, x): member in TestFX.test_trace_dict_proxy_keys.CallsModWithDict
2555 def forward(self, x): member in TestFX.test_module_deepcopy_edit_nodes.Foo
2578 def forward(self, x): member in TestFX.test_direct_param_use.TransposeTest
2586 def forward(self, x): member in TestFX.test_direct_param_use.Foo
2594 def forward(self, y=1): member in TestFX.test_single_default_arg.M
2603 def forward(self, y=1, z=2): member in TestFX.test_multiple_default_args.M
2613 def forward(self, x, y=1): member in TestFX.test_regular_and_default_args.M
2622 def forward(self): member in TestFX.test_string_literal_return.M
2630 def forward(self, x): member in TestFX.test_namedtuple_return_qualname.NamedTupReturn
2652 def forward(self): member in TestFX.test_torchbind_class_attribute_in_fx.FooBar1234
2667 def forward(self, x): member in TestFX.test_torchbind_class_attribute_in_fx_tensor_arg.FooBar2341
2680 def forward(self, x): member in TestFX.test_script_method_trace.Scripted
2688 def forward(self, x): member in TestFX.test_script_method_trace.Holder
2700 def forward(self, x): member in TestFX.test_namedtuple_return_trace.NamedTupReturn
2709 def forward(self, inp): member in TestFX.test_named_tuple_inlined.NamedTupMod
2746 def forward(self, x: List[str]) -> List[str]: member in TestFX.test_return_type_exists.ReturnTypeModule
2761 def forward(self, x): member in TestFX.getitem_inner.GetItem1
2765 def forward(self, x): member in TestFX.getitem_inner.GetItem2
2769 def forward(self, x): member in TestFX.getitem_inner.GetItem3
2802 def forward(self, x): member in TestFX.test_user_friendly_call_provenance_with_module.M
2822 def forward(self, x): member in TestFX.test_snake_case.M
2855 def forward(self, x, y): member in TestFX.test_partial_trace.Foo
2884 def forward(self, x): member in TestFX.test_custom_traceback_raised_when_exception_source_is_graphmodule.M
2911 def forward(self, x): member in TestFX.test_custom_traceback_not_raised_when_exception_source_is_submodule.M
2929 def forward(self, x): member in TestFX.test_graph_module_replicate_for_dp.Foo
2944 def forward(self, x: torch.Tensor, y: int, z: int): member in TestFX.test_ast_rewriter_rewrites_assert.M
2956 def forward(self, x: torch.Tensor, y: int, z: int): member in TestFX.test_ast_rewriter_rewrites_assert_with_message.M
2985 def forward(self, x: torch.Tensor): member in TestFX.test_ast_rewriter_reassigns_submodules.M
3057 def forward(self, x): member in TestFX.test_ast_rewriter_wrapped_via_decorator.F
3096 def forward(self, x: torch.Tensor): member in TestFX.test_ast_rewriter_wrap_with_submodule.M
3116 def forward(self, x): member in TestFX.test_submodule_manipulation_API.C
3126 def forward(self, x): member in TestFX.test_submodule_manipulation_API.B
3135 def forward(self, x): member in TestFX.test_submodule_manipulation_API.A
3262 def forward(self, x): member in TestFX.test_delete_unused_submodules_leaf.SubModule
3272 def forward(self, x): member in TestFX.test_delete_unused_submodules_leaf.Model
3295 def forward(self, x): member in TestFX.test_fx_stateless.MockModule
3319 def forward(self, t): member in TestFX.test_tracing_graphmodules_as_leaf_submodules.A
3328 def forward(self, t): member in TestFX.test_tracing_graphmodules_as_leaf_submodules.B
3346 def forward(self, t): member in TestFX.test_tracing_graphmodules_as_leaf_submodules.M
3445 def forward(self, x): member in TestFX._test_graph_module_init_buffer_param_copied.MyModule
3490 def forward(self, x: torch.Tensor, a: A) -> torch.Tensor: member in TestFX.test_annotations_with_no_forward_references.M
3501 def forward(self, x: 'torch.Tensor', a: 'A') -> 'torch.Tensor': member in TestFX.test_annotations_with_forward_references.M
3512 def forward(self, x: List[torch.Tensor], a: A) -> torch.Tensor: member in TestFX.test_annotations_with_non_torch_reference_and_no_internal_forward_references.M
3523 def forward(self, x: List['torch.Tensor'], a: A) -> 'torch.Tensor': member in TestFX.test_annotations_with_non_torch_reference_and_internal_forward_references.M
3539 def forward(self, x: Tuple[()], y: Tuple[str, Tuple[()]]): member in TestFX.test_annotations_empty_tuple.Foo
4230 def forward(self, x): member in TestFXAPIBackwardCompatibility.test_adding_side_effect_function.TestModule