Searched defs:FakeScriptObject (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_export/passes/ |
| H A D | lift_constants_pass.py | 40 self, key: Union[torch.Tensor, torch.ScriptObject, FakeScriptObject] argument
54 self, key: Union[torch.Tensor, torch.ScriptObject, FakeScriptObject], value: Any argument
105 key: Union[torch.Tensor, torch.ScriptObject, FakeScriptObject], argument
|
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | _trace.py | 350 constants: Dict[str, Union[torch.Tensor, FakeScriptObject, torch.ScriptObject]], argument
388 constants: Dict[str, Union[torch.Tensor, FakeScriptObject, torch.ScriptObject]], argument
840 constants: Dict[str, Union[torch.Tensor, FakeScriptObject, torch.ScriptObject]], argument
861 constants: Dict[str, Union[torch.Tensor, FakeScriptObject, torch.ScriptObject]], argument
|
| H A D | exported_program.py | 673 Dict[str, Union[torch.Tensor, FakeScriptObject, torch._C.ScriptObject]] argument
|
| /aosp_15_r20/external/pytorch/torch/_library/ |
| H A D | fake_class_registry.py | 12 class FakeScriptObject: class
24 self_fake_obj: FakeScriptObject,
|