• Home
  • Raw
  • Download

Lines Matching defs:forward

56             def forward(self, x):  member in TestSerialize.test_export_with_extension_op_serialization.TestModule
127 def forward(self, x): member in TestSerialize.test_predispatch_export_with_autograd_op.Foo
155 def forward(self, x, y, use_p=False): member in TestSerialize.test_export_example_inputs_preserved.MyModule
189 def forward(self, x): member in TestSerialize.test_metadata_parsing_with_layer_split.MyModule
215 def forward(self, x): member in TestSerialize.test_serialize_constant_outputs.MyModule
237 def forward(self, x, w, b): member in TestSerialize.test_serialize_multiple_returns_from_node.MyModule
273 def forward(self, a, b, c) -> torch.Tensor: member in TestSerialize.test_serialize_sym_int.DynamicShapeSimpleModel
307 def forward(self, x): member in TestSerialize.test_serialize_list_returns.MyModule
349 def forward(self, x): member in TestSerialize.test_multi_return_some_unused.MyModule
371 def forward(self, x): member in TestSerialize.test_rational_ranges.M
399 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestSerialize.test_kwargs_default.Foo
419 def forward(self, x: torch.Tensor, y: torch.Tensor) -> torch.Tensor: member in TestSerialize.test_canonicalize.Module
435 def forward(self, x): member in TestSerialize.test_int_list.M
622 def forward(self, a, b, c): member in TestDeserialize.test_optional_tuple.M
670 def forward(self, x, y, z, n): member in TestDeserialize.test_auto_functionalize.M
693 def forward(self, x, w, b): member in TestDeserialize.test_multi_return.MyModule
714 def forward(self, x): member in TestDeserialize.test_basic.MyModule
728 def forward(self, a, b, c) -> torch.Tensor: member in TestDeserialize.test_dynamic.DynamicShapeSimpleModel
743 def forward(self, x, y): member in TestDeserialize.test_sym_bool.Module
752 def forward(self, x): member in TestDeserialize.test_shape.Foo
769 def forward(self, x): member in TestDeserialize.test_module.M
785 def forward(self, x): member in TestDeserialize.test_module_meta.M
800 def forward(self, x, y): member in TestDeserialize.test_cond.M
818 def forward(self, xs, y): member in TestDeserialize.test_map.Module
847 def forward(self, x, y): member in TestDeserialize.test_tensor_tensor_list.M
860 def forward(self, x, y, z): member in TestDeserialize.test_list_of_optional_tensors.MyModule
870 def forward(self, x): member in TestDeserialize.test_sym_ite.Foo
880 def forward(self, x): member in TestDeserialize.test_multiple_getitem.M
933 def forward(self, x, y): member in TestDeserialize.test_constraints.Module
943 def forward(self, x): member in TestDeserialize.test_get_attr.Module
951 def forward(self, x): member in TestDeserialize.test_get_attr_list.Module
965 def forward(self, x): member in TestDeserialize.test_device.MyModule
981 def forward(self, x): member in TestDeserialize.test_custom_obj_tuple_out.MyModule
997 def forward(self, x): member in TestDeserialize.test_custom_obj.MyModule
1012 def forward(self, x): member in TestDeserialize.test_custom_obj_list_out.MyModule
1028 def forward(self): member in TestDeserialize.test_export_no_inputs.M
1044 def forward(self, x): member in TestSchemaVersioning.test_error.Module
1077 def forward(self, x): member in TestSaveLoad.test_save_buffer.Module
1095 def forward(self, x): member in TestSaveLoad.test_save_file.Foo
1112 def forward(self, x, y): member in TestSaveLoad.test_save_path.Foo
1131 def forward(self, x): member in TestSaveLoad.test_save_extra.Foo
1149 def forward(self, x): member in TestSaveLoad.test_version_error.Foo
1176 def forward(self, x): member in TestSaveLoad.test_save_constants.Foo
1200 def forward(self, x): member in TestSerializeCustomClass.test_custom_class.Foo
1252 def forward(self, x): member in TestSerializeCustomClass.test_custom_class_containing_fake_tensor.Foo
1268 def forward(self, x): member in TestSerializeCustomClass.test_custom_tag_metadata_serialization.Foo
1303 def forward(self, x): member in TestSerializeCustomClass.test_custom_tag_metadata_decomp.Foo
1370 def forward(self, x): member in TestSerializeCustomClass.test_custom_tag_metadata_copy.Foo