Home
last modified time | relevance | path

Searched refs:FakeScriptObject (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_export/passes/
H A Dlift_constants_pass.py9 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 Dfake_class_registry.py12 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 Dnon_strict_utils.py21 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 Dgraph_signature.py6 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.py60 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 Dexported_program.py27 from torch._library.fake_class_registry import FakeScriptObject
673 Dict[str, Union[torch.Tensor, FakeScriptObject, torch._C.ScriptObject]] argument
H A Dunflatten.py14 from torch._library.fake_class_registry import FakeScriptObject
80 from_obj, FakeScriptObject
/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A Dtorchbind.py9 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.py29 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 Dutils.py15 from torch._library.fake_class_registry import FakeScriptObject
30 FakeScriptObject,
/aosp_15_r20/external/executorch/exir/
H A Dlowered_backend_module.py28 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 Daot_autograd.py594 FakeScriptObject,
604 if isinstance(arg, FakeScriptObject)
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dproxy_tensor.py49 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 Dtest_torchbind.py12 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.py996 obj, torch._library.fake_class_registry.FakeScriptObject
/aosp_15_r20/external/pytorch/torch/_export/serde/
H A Dserialize.py41 from torch._library.fake_class_registry import FakeScriptObject
1424 constants: Dict[str, Union[torch.Tensor, FakeScriptObject, torch.ScriptObject]]