Home
last modified time | relevance | path

Searched refs:ScriptObject (Results 1 – 25 of 58) sorted by relevance

123

/aosp_15_r20/external/pytorch/torch/_export/passes/
H A Dlift_constants_pass.py37 self._script_object_map: Dict[int, torch.ScriptObject] = {}
40 self, key: Union[torch.Tensor, torch.ScriptObject, FakeScriptObject] argument
42 real_key = hash(key) if isinstance(key, torch.ScriptObject) else key
46 def __setitem__(self, key: Union[torch.Tensor, torch.ScriptObject], value): argument
54 self, key: Union[torch.Tensor, torch.ScriptObject, FakeScriptObject], value: Any argument
56 if isinstance(key, torch.ScriptObject):
71 real_key = hash(key) if isinstance(key, torch.ScriptObject) else key
86 real_key = hash(key) if isinstance(key, torch.ScriptObject) else key
105 key: Union[torch.Tensor, torch.ScriptObject, FakeScriptObject], argument
115 ) -> Dict[str, Union[torch.Tensor, torch.ScriptObject, FakeScriptObject]]:
[all …]
H A Dreplace_quantized_ops_with_standard_ops_pass.py156 ) -> torch._C.ScriptObject:
165 assert isinstance(mod, torch._C.ScriptObject)
628 if isinstance(v, torch.ScriptObject):
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/include/lldb/API/
DSBScriptObject.h15 class ScriptObject; variable
43 lldb_private::ScriptObject *get();
45 lldb_private::ScriptObject &ref();
47 const lldb_private::ScriptObject &ref() const;
50 std::unique_ptr<lldb_private::ScriptObject> m_opaque_up;
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/include/lldb/API/
DSBScriptObject.h15 class ScriptObject; variable
43 lldb_private::ScriptObject *get();
45 lldb_private::ScriptObject &ref();
47 const lldb_private::ScriptObject &ref() const;
50 std::unique_ptr<lldb_private::ScriptObject> m_opaque_up;
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r522817/include/lldb/API/
DSBScriptObject.h15 class ScriptObject; variable
43 lldb_private::ScriptObject *get();
45 lldb_private::ScriptObject &ref();
47 const lldb_private::ScriptObject &ref() const;
50 std::unique_ptr<lldb_private::ScriptObject> m_opaque_up;
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/include/lldb/API/
DSBScriptObject.h15 class ScriptObject; variable
43 lldb_private::ScriptObject *get();
45 lldb_private::ScriptObject &ref();
47 const lldb_private::ScriptObject &ref() const;
50 std::unique_ptr<lldb_private::ScriptObject> m_opaque_up;
/aosp_15_r20/external/pytorch/torch/_library/
H A Dfake_class_registry.py13 def __init__(self, wrapped_obj: Any, script_class_name: str, x: torch.ScriptObject): argument
40 def from_real(cls, real_obj: torch.ScriptObject): argument
102 def tracing_with_real(x: torch.ScriptObject) -> bool: argument
114 fake_mode, x: torch.ScriptObject argument
115 ) -> Union[FakeScriptObject, torch.ScriptObject]:
287 def _find_fake_class_for_script_object(x: torch.ScriptObject) -> Any: argument
/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A Dtorchbind.py46 if isinstance(self.raw_owner, torch.ScriptObject):
58 KNOWN_TYPES.append(torch.ScriptObject)
63 KNOWN_TYPES.pop() is torch.ScriptObject
70 if isinstance(obj, torch.ScriptObject):
92 if isinstance(obj, torch.ScriptObject):
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r522817/include/lldb/Interpreter/
DScriptObject.h15 class ScriptObject {
17 ScriptObject(lldb::ScriptObjectPtr ptr, lldb::ScriptLanguage lang) in ScriptObject() function
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/include/lldb/Interpreter/
DScriptObject.h15 class ScriptObject {
17 ScriptObject(lldb::ScriptObjectPtr ptr, lldb::ScriptLanguage lang) in ScriptObject() function
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/include/lldb/Interpreter/
DScriptObject.h15 class ScriptObject {
17 ScriptObject(lldb::ScriptObjectPtr ptr, lldb::ScriptLanguage lang) in ScriptObject() function
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/include/lldb/Interpreter/
DScriptObject.h15 class ScriptObject {
17 ScriptObject(lldb::ScriptObjectPtr ptr, lldb::ScriptLanguage lang) in ScriptObject() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dmodule_python.h21 static py::handle ScriptObject = in as_object() local
23 if (py::isinstance(obj, ScriptObject)) { in as_object()
/aosp_15_r20/external/pytorch/torch/_export/
H A Dnon_strict_utils.py91 if _is_constant_argument(t) or isinstance(t, torch.ScriptObject):
399 torch.ScriptObject,
464 if isinstance(obj, torch.ScriptObject):
477 torch.ScriptObject, _maybe_fakify_obj, (args, kwargs)
H A Dconverter.py456 and isinstance(self.name_to_constant[fqn], torch.ScriptObject)
537 if isinstance(self.name_to_constant[value_name], torch.ScriptObject):
613 self.name_to_constant[name], torch.ScriptObject
1510 if isinstance(v, (torch.Tensor, torch.ScriptObject))
1575 elif isinstance(value, torch.ScriptObject):
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_torchbind.py34 if isinstance(a, torch.ScriptObject) and isinstance(b, torch.ScriptObject):
39 def _check_script_obj_equal(test_case, a: torch.ScriptObject, b: torch.ScriptObject): argument
52 if isinstance(a, torch.ScriptObject) and isinstance(b, torch.ScriptObject):
728 def forward(self, safe_obj: torch.ScriptObject) -> None: argument
992 def forward(self, tq: torch.ScriptObject, x: torch.Tensor) -> None: argument
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dtorchbind_impls.py10 _TENSOR_QUEUE_GLOBAL_TEST: Optional[torch.ScriptObject] = None
26 def _empty_tensor_queue() -> torch.ScriptObject:
/aosp_15_r20/external/pytorch/torch/_C/
H A D_distributed_c10d.pyi9 from torch._C import ScriptObject
269 def boxed(self) -> ScriptObject: ...
271 def unbox(obj: ScriptObject) -> Work: ...
501 def boxed(self) -> ScriptObject: ...
503 def unbox(obj: ScriptObject) -> ProcessGroup: ...
/aosp_15_r20/external/pytorch/torch/ao/quantization/fx/
H A Dgraph_module.py136 getattr(self, attr_name), torch._C.ScriptObject
178 …with("_packed_weight") and isinstance(state_dict[attr_name], torch._C.ScriptObject): # type: igno…
/aosp_15_r20/external/executorch/exir/verification/
H A Dverifier.py62 torch.ScriptObject,
222 torch.ScriptObject,
/aosp_15_r20/external/pytorch/torch/fx/
H A D_symbolic_trace.py28 from torch._C import ScriptObject # type: ignore[attr-defined]
400 if isinstance(a, (torch.Tensor, ScriptObject, FakeScriptObject)):
759 ScriptObject,
766 if isinstance(v, (torch.Tensor, ScriptObject, FakeScriptObject)):
/aosp_15_r20/external/pytorch/torch/export/
H A Dunflatten.py57 from_obj: Union[torch.Tensor, torch.ScriptObject], argument
86 torch.ScriptObject,
308 id_to_const: Dict[int, Union[torch.Tensor, torch._C.ScriptObject]] = {}
1235 assert isinstance(state_attr, (torch.Tensor, torch.ScriptObject))
H A D_trace.py122 torch.ScriptObject,
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 Dgraph_signature.py457 from torch import ScriptObject, SymInt
475 elif isinstance(val, ScriptObject):
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dscript_object.py32 return issubclass(user_cls, torch.ScriptObject)

123