/aosp_15_r20/external/pytorch/torch/_export/passes/ |
H A D | lift_constants_pass.py | 9 from torch._library.fake_class_registry import FakeScriptObject 32 Union[int, torch.Tensor, FakeScriptObject], List[Any] 40 self, key: Union[torch.Tensor, torch.ScriptObject, FakeScriptObject] argument 43 assert isinstance(real_key, (int, torch.Tensor, FakeScriptObject)) 54 self, key: Union[torch.Tensor, torch.ScriptObject, FakeScriptObject], value: Any argument 61 elif isinstance(key, (torch.Tensor, FakeScriptObject)): 105 key: Union[torch.Tensor, torch.ScriptObject, FakeScriptObject], argument 115 ) -> Dict[str, Union[torch.Tensor, torch.ScriptObject, FakeScriptObject]]: 134 str, Union[torch.Tensor, torch.ScriptObject, FakeScriptObject] 178 if isinstance(constant_val, (torch.ScriptObject, FakeScriptObject)): [all …]
|
/aosp_15_r20/external/pytorch/torch/_library/ |
H A D | fake_class_registry.py | 12 class FakeScriptObject: class 24 self_fake_obj: FakeScriptObject, argument 115 ) -> Union[FakeScriptObject, torch.ScriptObject]: 141 …fake_x_wrapped = FakeScriptObject(fake_x, x._type().qualified_name(), x) # type: ignore[attr-defi…
|
/aosp_15_r20/external/pytorch/torch/_export/ |
H A D | non_strict_utils.py | 21 from torch._library.fake_class_registry import FakeScriptObject 400 FakeScriptObject, 438 isinstance(obj, FakeScriptObject) for obj in constant_attrs.values() 441 lambda obj: isinstance(obj, FakeScriptObject), (args, kwargs)
|
/aosp_15_r20/external/pytorch/torch/export/ |
H A D | graph_signature.py | 6 from torch._library.fake_class_registry import FakeScriptObject 46 fake_val: Optional[FakeScriptObject] = None 458 from torch._library.fake_class_registry import FakeScriptObject 477 elif isinstance(val, FakeScriptObject):
|
H A D | _trace.py | 60 from torch._library.fake_class_registry import FakeScriptObject 121 FakeScriptObject, 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 1694 fqn: map_fake_to_real[obj] if isinstance(obj, FakeScriptObject) else obj
|
H A D | exported_program.py | 27 from torch._library.fake_class_registry import FakeScriptObject 673 Dict[str, Union[torch.Tensor, FakeScriptObject, torch._C.ScriptObject]] argument
|
H A D | unflatten.py | 14 from torch._library.fake_class_registry import FakeScriptObject 80 from_obj, FakeScriptObject
|
/aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
H A D | torchbind.py | 9 from torch._library.fake_class_registry import _ns_and_class_name, FakeScriptObject 72 elif isinstance(obj, FakeScriptObject):
|
/aosp_15_r20/external/pytorch/torch/fx/ |
H A D | _symbolic_trace.py | 29 from torch._library.fake_class_registry import FakeScriptObject 400 if isinstance(a, (torch.Tensor, ScriptObject, FakeScriptObject)): 760 FakeScriptObject 766 if isinstance(v, (torch.Tensor, ScriptObject, FakeScriptObject)):
|
/aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
H A D | utils.py | 15 from torch._library.fake_class_registry import FakeScriptObject 30 FakeScriptObject,
|
/aosp_15_r20/external/executorch/exir/ |
H A D | lowered_backend_module.py | 28 from torch._library.fake_class_registry import FakeScriptObject 443 Dict[str, Union[torch.Tensor, FakeScriptObject, torch.ScriptObject]],
|
/aosp_15_r20/external/pytorch/torch/_functorch/ |
H A D | aot_autograd.py | 594 FakeScriptObject, 604 if isinstance(arg, FakeScriptObject)
|
/aosp_15_r20/external/pytorch/torch/fx/experimental/ |
H A D | proxy_tensor.py | 49 from torch._library.fake_class_registry import FakeScriptObject 104 _AnyScriptObject = (torch.ScriptObject, FakeScriptObject) 105 _AnyScriptObjectType = Union[torch.ScriptObject, FakeScriptObject] 1990 x, FakeScriptObject
|
/aosp_15_r20/external/pytorch/test/export/ |
H A D | test_torchbind.py | 12 from torch._library.fake_class_registry import FakeScriptObject 569 test.assertTrue(isinstance(tq, FakeScriptObject)) 628 self.current_test.assertTrue(isinstance(tq, FakeScriptObject))
|
/aosp_15_r20/external/pytorch/torch/ |
H A D | _ops.py | 996 obj, torch._library.fake_class_registry.FakeScriptObject
|
/aosp_15_r20/external/pytorch/torch/_export/serde/ |
H A D | serialize.py | 41 from torch._library.fake_class_registry import FakeScriptObject 1424 constants: Dict[str, Union[torch.Tensor, FakeScriptObject, torch.ScriptObject]]
|