Home
last modified time | relevance | path

Searched defs:VariableTracker (Results 1 – 22 of 22) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dhigher_order_ops.py133 def _check_supported_callable_arg(tx, func_var: VariableTracker, arg_name):
565 self, tx, args: List[VariableTracker], kwargs: Dict[str, VariableTracker]
575 self, tx, args: "List[VariableTracker]", kwargs: "Dict[str, VariableTracker]"
762 self, tx, args: List[VariableTracker], kwargs: Dict[str, VariableTracker]
788 self, tx, args: List[VariableTracker], kwargs: Dict[str, VariableTracker]
942 self, tx, args: List[VariableTracker], kwargs: Dict[str, VariableTracker]
1045 self, tx, args: List[VariableTracker], kwargs: Dict[str, VariableTracker]
1132 self, tx, args: "List[VariableTracker]", kwargs: "Dict[str, VariableTracker]"
1179 self, tx, args: "List[VariableTracker]", kwargs: "Dict[str, VariableTracker]"
1243 self, tx, args: "List[VariableTracker]", kwargs: "Dict[str, VariableTracker]"
[all …]
H A Ddicts.py131 self, items: Dict[VariableTracker, VariableTracker], user_cls=dict, **kwargs argument
220 self, tx: "InstructionTranslator", arg: VariableTracker
227 def getitem_const(self, tx: "InstructionTranslator", arg: VariableTracker):
233 def maybe_getitem_const(self, arg: VariableTracker):
243 args: "List[VariableTracker]",
398 args: "List[VariableTracker]",
427 items: List[VariableTracker],
465 args: List[VariableTracker],
540 def getitem_const(self, tx: "InstructionTranslator", arg: VariableTracker):
547 items: List[VariableTracker],
[all …]
H A Ddistributed.py121 args: "List[VariableTracker]",
162 args: "List[VariableTracker]",
228 args: "List[VariableTracker]",
270 args: "List[VariableTracker]",
312 module: VariableTracker,
313 user_hooks: VariableTracker,
314 user_pre_hooks: VariableTracker,
363 module: VariableTracker,
364 user_hooks: VariableTracker,
365 user_pre_hooks: VariableTracker,
[all …]
H A Dmisc.py155 args: "List[VariableTracker]",
297 args: "List[VariableTracker]",
392 def __init__(self, inspected: VariableTracker, **kwargs) -> None:
426 args: "List[VariableTracker]",
507 defaults: Dict[str, VariableTracker], argument
578 args: "List[VariableTracker]",
754 args: "List[VariableTracker]",
878 args: "List[VariableTracker]",
942 args: "List[VariableTracker]",
974 args: "List[VariableTracker]",
[all …]
H A Dfunctions.py88 self, tx, args: "List[VariableTracker]", kwargs: "Dict[str, VariableTracker]"
286 self, tx, args: "List[VariableTracker]", kwargs: "Dict[str, VariableTracker]"
313 self, tx, args: "List[VariableTracker]", kwargs: "Dict[str, VariableTracker]"
359 self, tx, args: "List[VariableTracker]", kwargs: "Dict[str, VariableTracker]"
376 self, tx, args: "List[VariableTracker]", kwargs: "Dict[str, VariableTracker]"
601 self, tx, args: "List[VariableTracker]", kwargs: "Dict[str, VariableTracker]"
726 self, tx, args: "List[VariableTracker]", kwargs: "Dict[str, VariableTracker]"
765 def __init__(self, func: VariableTracker, args, keywords, **kwargs):
792 self, tx, args: "List[VariableTracker]", kwargs: "Dict[str, VariableTracker]"
869 self, tx, args: "List[VariableTracker]", kwargs: "Dict[str, VariableTracker]"
[all …]
H A Dlists.py62 items: List[VariableTracker],
90 def getitem_const(self, tx: "InstructionTranslator", arg: VariableTracker):
116 args: List["VariableTracker"],
284 def getitem_const(self, tx: "InstructionTranslator", arg: VariableTracker):
323 args: List["VariableTracker"],
413 args: List["VariableTracker"],
481 args: List["VariableTracker"],
545 args: List["VariableTracker"],
576 items: List[VariableTracker],
679 args: List["VariableTracker"],
[all …]
H A Diter.py42 args: "List[VariableTracker]",
224 def __init__(self, item: VariableTracker, **kwargs) -> None:
280 saved: List[VariableTracker] = None,
282 item: Optional[VariableTracker] = None,
334 iterables: List[Union[List[VariableTracker], VariableTracker]], argument
447 fn: VariableTracker,
448 iterables: List[Union[List[VariableTracker], VariableTracker]], argument
H A Dbuiltin.py953 args: "List[VariableTracker]",
973 args: "List[VariableTracker]",
1171 def call_abs(self, tx: "InstructionTranslator", arg: "VariableTracker"):
1178 def call_pos(self, tx: "InstructionTranslator", arg: "VariableTracker"):
1185 def call_index(self, tx: "InstructionTranslator", arg: "VariableTracker"):
1550 def call_next(self, tx: "InstructionTranslator", arg: VariableTracker):
1634 obj: VariableTracker,
1635 name_var: VariableTracker,
1762 obj: VariableTracker,
1763 name_var: VariableTracker,
[all …]
H A Duser_defined.py303 args: "List[VariableTracker]",
340 args: "List[VariableTracker]",
699 args: "List[VariableTracker]",
844 args: "List[VariableTracker]",
1275 args: "List[VariableTracker]",
1296 args: "List[VariableTracker]",
H A Dnn_module.py344 args: "List[VariableTracker]",
452 args: "List[VariableTracker]",
853 args: "List[VariableTracker]",
963 args: "List[VariableTracker]",
H A Dtensor.py510 args: "List[VariableTracker]",
962 def _method_register_hook(self, name: str, hook: VariableTracker):
1129 args: "List[VariableTracker]",
1233 args: "List[VariableTracker]",
1311 args: List[VariableTracker],
1353 args: List[VariableTracker],
H A Dbase.py118 class VariableTracker(metaclass=VariableTrackerMeta): class
145 fn: Callable[["VariableTracker"], None],
321 args: "List[VariableTracker]",
330 args: "List[VariableTracker]",
H A Dctx_manager.py114 args: "List[VariableTracker]",
489 args: "List[VariableTracker]",
952 args: "List[VariableTracker]",
996 args: "List[VariableTracker]",
1065 args: "List[VariableTracker]",
1122 args: "List[VariableTracker]",
H A Dconstant.py102 def getitem_const(self, tx: "InstructionTranslator", arg: VariableTracker):
139 args: "List[VariableTracker]",
H A Dtorch.py215 args: "List[VariableTracker]",
830 args: "List[VariableTracker]",
H A Doptimizer.py64 args: "List[VariableTracker]",
H A Dtorch_function.py361 args: "List[VariableTracker]",
H A Dlazy.py137 self, sym_node_variable: SymNodeVariable, fmt_spec_var: VariableTracker
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dside_effects.py160 def store_attr(self, item: VariableTracker, name: str, value: VariableTracker):
183 def load_global(self, gvar: VariableTracker, name: str):
187 def store_global(self, gvar: VariableTracker, name: str, value: VariableTracker):
222 variable: VariableTracker,
247 variable: VariableTracker,
350 def visit(var: VariableTracker):
368 def is_live(var: Union[MutableLocalBase, VariableTracker]): argument
H A Dsymbolic_convert.py812 fn: VariableTracker,
813 args: List[VariableTracker],
1001 def push(self, val: Optional[VariableTracker], name: Any = None):
1010 def push_many(self, vals: List[VariableTracker]):
2520 def strict_translation_mode(self, check_fn: Callable[[VariableTracker], bool]):
2549 symbolic_locals: Dict[str, VariableTracker], argument
3047 parent, func: VariableTracker, args: List[VariableTracker], kwargs
3180 symbolic_locals: Dict[str, VariableTracker], argument
H A Doutput_graph.py441 def add_backward_state_hook(self, hook: VariableTracker, prefix="hook"):
H A Dutils.py2620 def has_torch_function(vt: "torch._dynamo.variables.base.VariableTracker") -> bool: