Lines Matching defs:forward
140 def forward(self, x, y): member in TestConverter.test_ts2ep_converter_basic.MSingle
144 def forward(self, x, y): member in TestConverter.test_ts2ep_converter_basic.MMulti
156 def forward(self, x: torch.Tensor, y: torch.Tensor): member in TestConverter.test_ts2ep_converter_container_output.MOutputList
163 def forward(self, x: torch.Tensor, y: torch.Tensor): member in TestConverter.test_ts2ep_converter_container_output.MOutputTuple
170 def forward(self, x: torch.Tensor, y: torch.Tensor): member in TestConverter.test_ts2ep_converter_container_output.MOutputDict
184 def forward(self, x): member in TestConverter.test_aten_dim.Module
193 def forward(self, x: torch.Tensor): member in TestConverter.test_aten_len.Module
202 def forward(self, x: List[int]): member in TestConverter.test_aten_len.Module
211 def forward(self, x: Dict[int, str]): member in TestConverter.test_aten_len.Module
220 def forward(self, x: Dict[bool, str]): member in TestConverter.test_aten_len.Module
229 def forward(self, x: Dict[float, str]): member in TestConverter.test_aten_len.Module
238 def forward(self, x: Dict[torch.Tensor, str]): member in TestConverter.test_aten_len.Module
256 def forward(self, x: List[torch.Tensor]): member in TestConverter.test_aten_add_t.Module
269 def forward(self, x: torch.Tensor, y: torch.Tensor): member in TestConverter.test_aten_to_dtype_with_mutating_storage.Module
279 def forward(self, x: torch.Tensor, y: torch.Tensor) -> torch.Tensor: member in TestConverter.test_prim_min.Module
315 def forward(self, x: torch.Tensor, y: torch.Tensor) -> torch.Tensor: member in TestConverter.test_prim_max.Module
351 def forward(self, x): member in TestConverter.test_aten___getitem___list.Module
360 def forward(self, x): member in TestConverter.test_aten___getitem___dict.Module
373 def forward(self, x): member in TestConverter.test_prim_device.Module
383 def forward(self, x): member in TestConverter.test_prim_device_cuda.Module
392 def forward(self, x): member in TestConverter.test_prim_dtype.Module
413 def forward(self, x: torch.Tensor, y: torch.Tensor): member in TestConverter.test_convert_if_basic.M
436 def forward(self, x: torch.Tensor, y: torch.Tensor): member in TestConverter.test_convert_if_tuple_out.M
463 def forward(self, x: torch.Tensor, y: torch.Tensor): member in TestConverter.test_convert_if_multiple_out.M
486 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestConverter.test_profiler__record_function.Module
497 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestConverter.test_aten_floordiv.Module
505 def forward( member in TestConverter.test_aten___is__.Module
517 def forward( member in TestConverter.test_aten___isnot__.Module
529 def forward( member in TestConverter.test_aten___not__.Module
541 def forward(self, x): member in TestConverter.test_ts2ep_converter_unpack.MUnpackList
546 def forward(self, x_tuple: Tuple[torch.Tensor, torch.Tensor]): member in TestConverter.test_ts2ep_converter_unpack.MUnpackTuple
567 def forward(self, x, y): member in TestConverter.test_convert_retrace_nested_scripted_modules.Wrapper
575 def forward(self, x, y): member in TestConverter.test_convert_retrace_nested_scripted_modules.LinearM
586 def forward(self, x: torch.Tensor, y: torch.Tensor): member in TestConverter.test_convert_retrace_nested_scripted_modules.M
600 def forward(self, x: torch.Tensor, y: torch.Tensor): member in TestConverter.test_convert_retrace_nested_scripted_modules.NestedM
618 def forward(self, x: torch.Tensor): member in TestConverter.test_convert_nn_module_with_nested_param.M
627 def forward(self, x: torch.Tensor): member in TestConverter.test_convert_nn_module_with_nested_param.NestedM
636 def forward(self, x: torch.Tensor): member in TestConverter.test_convert_nn_module_with_nested_param.SuperNestedM
651 def forward(self, x: torch.Tensor): member in TestConverter.test_convert_nn_module_with_nested_buffer.M
660 def forward(self, x: torch.Tensor): member in TestConverter.test_convert_nn_module_with_nested_buffer.NestedM
669 def forward(self, x: torch.Tensor): member in TestConverter.test_convert_nn_module_with_nested_buffer.SuperNestedM
685 def forward(self, x: torch.Tensor): member in TestConverter.test_convert_nn_module_with_nested_if_and_buffer.M
695 def forward(self, x: torch.Tensor): member in TestConverter.test_convert_nn_module_with_nested_if_and_buffer.NestedM
710 def forward(self, x: torch.Tensor): member in TestConverter.test_convert_nn_module_with_nested_if_and_buffer.SuperNestedM
740 def forward(self, x: torch.Tensor): member in TestConverter.test_convert_nn_module_with_nested_if_and_param.M
750 def forward(self, x: torch.Tensor): member in TestConverter.test_convert_nn_module_with_nested_if_and_param.NestedM
765 def forward(self, x: torch.Tensor): member in TestConverter.test_convert_nn_module_with_nested_if_and_param.SuperNestedM1
780 def forward(self, x: torch.Tensor): member in TestConverter.test_convert_nn_module_with_nested_if_and_param.SuperNestedM2
843 def forward(self, x: torch.Tensor): member in TestConverter.test_ts2ep_converter_contains.MIn
847 def forward(self, x: torch.Tensor): member in TestConverter.test_ts2ep_converter_contains.MNotIn
851 def forward(self, x: torch.Tensor, x_dict: Dict[torch.Tensor, str]): member in TestConverter.test_ts2ep_converter_contains.MTensorIn
894 def forward(self, x): member in TestConverter.test_ts2ep_converter_custom_op.M
966 def forward(self): member in TestConverter.test_implicit_constant_to_tensor_handling.M1
970 def forward(self, x): member in TestConverter.test_implicit_constant_to_tensor_handling.M2
1009 def forward(self, x): member in TestConverter.test_aten_tensor_dtype_int.M
1019 def forward(self, x): member in TestConverter.test_aten_tensor_prim_dtype.M
1029 def forward(self, x): member in TestConverter.test_aten_tensor_dynamic.M
1046 def forward(self, x): member in TestConverter.test_aten_tensor_dynamic.M
1065 def forward(self, x: torch.Tensor) -> List[int]: member in TestConverter.test_prim_tolist.Module
1072 def forward(self, x: torch.Tensor) -> List[List[int]]: member in TestConverter.test_prim_tolist.Module
1086 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestConverter.test_get_tensor_constants.Foo
1095 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestConverter.test_get_tensor_constants.Goo
1108 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestConverter.test_prim_SetAttr.Module
1122 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestConverter.test_prim_SetAttr.Module
1140 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestConverter.test_prim_SetAttr.Module
1158 def forward(self, x: torch.Tensor) -> torch.Tensor: member in TestConverter.test_prim_SetAttr.Module
1173 def forward(self, x): member in TestConverter.test_prim_SetAttr.M
1189 def forward(self, x: torch.Tensor): member in TestConverter.test_prim_SetAttr.M
1198 def forward(self, x: torch.Tensor, y: int) -> torch.Tensor: member in TestConverter.test_raise_exception.Module
1215 def forward(self, x: torch.Tensor, y: int) -> torch.Tensor: member in TestConverter.test_raise_exception.Module
1250 def forward(self, x, y): member in TestConverter.test_context_manager.M
1287 def forward(self, x: torch.Tensor, y: torch.Tensor): member in TestConverter.test_ts2ep_multi_outputs_on_call_ops.M
1300 def forward(self, x: List[torch.Tensor]): member in TestConverter.test_aten_append_t.M
1319 def forward(self, x: torch.Tensor): member in TestConverter.test_convert_script_object.M1
1380 def forward(self, x): member in TestConverter.test_ts2ep_convert_quantized_model.Standalone
1419 def forward(self, x): member in TestConverter.test_ts2ep_convert_quantized_model_with_opcontext.M