Home
last modified time | relevance | path

Searched defs:InstructionTranslator (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dbuiltin.py328 tx: "InstructionTranslator", a, b, *, forward_name=inplace_name
340 def dynamic_handler(tx: "InstructionTranslator", a, b, *, fn=op):
368 def tuple_add_handler(tx: "InstructionTranslator", a, b):
371 def size_add_handler(tx: "InstructionTranslator", a, b):
412 def list_iadd_handler(tx: "InstructionTranslator", a, b):
439 def expand_list_like(tx: "InstructionTranslator", lst, const):
459 def compare_by_value(tx: "InstructionTranslator", a, b):
469 def never(tx: "InstructionTranslator", a, b):
477 tx: "InstructionTranslator", a, b: ConstantVariable
483 tx: "InstructionTranslator", a: ConstantVariable, b
[all …]
H A Dtorch.py180 def call_hasattr(self, tx: "InstructionTranslator", name):
214 tx: "InstructionTranslator",
372 self, tx: "InstructionTranslator", *args, **kwargs
387 self, tx: "InstructionTranslator", *args, **kwargs
398 def handle_accumulate_grad_(self, tx: "InstructionTranslator", *args, **kwargs):
404 def handle_radians(self, tx: "InstructionTranslator", *args, **kwargs):
412 def handle_is_tensor(self, tx: "InstructionTranslator", arg):
426 def handle_is_floating_point(self, tx: "InstructionTranslator", input):
437 def handle_numel(self, tx: "InstructionTranslator", input):
445 def handle_tensor_size_rewrites(self, tx: "InstructionTranslator", input):
[all …]
H A Dctx_manager.py79 def set_cleanup_hook(self, tx: "InstructionTranslator", fn=None):
88 def exit(self, tx: "InstructionTranslator", *args):
113 tx: "InstructionTranslator",
161 def exit(self, tx: "InstructionTranslator", *args):
191 def create(tx: "InstructionTranslator", target_values, **kwargs):
216 def exit(self, tx: "InstructionTranslator", *args):
238 def create(tx: "InstructionTranslator", **kwargs):
260 def exit(self, tx: "InstructionTranslator", *args):
272 def create(tx: "InstructionTranslator", target_values, **kwargs):
295 def exit(self, tx: "InstructionTranslator", *args):
[all …]
H A Duser_defined.py156 def has_key_in_generic_dict(self, tx: "InstructionTranslator", key):
163 def var_getattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
235 def _call_cross_entropy_loss(self, tx: "InstructionTranslator", args, kwargs):
339 tx: "InstructionTranslator",
580 def call_hasattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
587 def const_getattr(self, tx: "InstructionTranslator", name):
661 def call_torch_function(self, tx: "InstructionTranslator", fn, types, args, kwargs):
798 def method_setattr_standard(self, tx: "InstructionTranslator", name, value):
843 tx: "InstructionTranslator",
973 def has_key_in_generic_dict(self, tx: "InstructionTranslator", key):
[all …]
H A Dmisc.py83 def _resolved_getattr_and_source(self, tx: "InstructionTranslator", name):
134 def var_getattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
284 def var_getattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
296 tx: "InstructionTranslator",
409 def var_getattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
478 def var_getattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
557 def var_getattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
644 def call_apply(self, tx: "InstructionTranslator", args, kwargs):
737 def call_backward(self, tx: "InstructionTranslator", args, kwargs):
747 def call_function(self, tx: "InstructionTranslator", args, kwargs):
[all …]
H A Dtorch_function.py88 def register_mutation(cls, tx: "InstructionTranslator"):
97 def register_device_context_insertion(cls, tx: "InstructionTranslator"):
111 def clear_default_device(cls, tx: "InstructionTranslator"):
178 def _get_subclass_type_var(tx: "InstructionTranslator", var):
191 def _is_attr_overidden(tx: "InstructionTranslator", var, name):
221 def build_torch_function_fn(tx: "InstructionTranslator", value, source):
233 def can_dispatch_torch_function(tx: "InstructionTranslator", args, kwargs):
239 def dispatch_torch_function(tx: "InstructionTranslator", fn, args, kwargs):
309 def var_getattr(self, tx: "InstructionTranslator", name):
346 def call_torch_function(self, tx: "InstructionTranslator", fn, types, args, kwargs):
H A Dnn_module.py54 def initialize_lazy_module(tx: "InstructionTranslator", mod, args, kwargs):
154 self, tx: "InstructionTranslator", source, submod, *key_extra, **options
190 def call_hasattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
214 def has_key_in_generic_dict(self, tx: "InstructionTranslator", key):
243 def var_getattr(self, tx: "InstructionTranslator", name):
852 tx: "InstructionTranslator",
904 self, tx: "InstructionTranslator", method, name, args, kwargs
1053 def getattr_helper(self, tx: "InstructionTranslator", field, name_vt):
1059 def var_getattr(self, tx: "InstructionTranslator", name):
1122 def manually_trace_nn_module_getattr(self, tx: "InstructionTranslator", name):
H A Ddicts.py220 self, tx: "InstructionTranslator", arg: VariableTracker
227 def getitem_const(self, tx: "InstructionTranslator", arg: VariableTracker):
540 def getitem_const(self, tx: "InstructionTranslator", arg: VariableTracker):
691 self, tx: "InstructionTranslator", name: str
899 def var_getattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
954 def var_getattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
965 def call_hasattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
991 tx: "InstructionTranslator",
1004 def _contains_helper(self, tx: "InstructionTranslator", key: VariableTracker):
1014 def call_contains(self, tx: "InstructionTranslator", key: VariableTracker):
[all …]
H A Dlists.py90 def getitem_const(self, tx: "InstructionTranslator", arg: VariableTracker):
148 def list_compare(tx: "InstructionTranslator", op, left, right):
284 def getitem_const(self, tx: "InstructionTranslator", arg: VariableTracker):
307 def var_getattr(self, tx: "InstructionTranslator", name):
449 def call_hasattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
560 def call_hasattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
692 def get_item_dyn(self, tx: "InstructionTranslator", arg: VariableTracker):
705 def call_hasattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
745 def var_getattr(self, tx: "InstructionTranslator", name):
769 def call_hasattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
[all …]
H A Ddistributed.py96 def var_getattr(self, tx: "InstructionTranslator", name: str) -> VariableTracker:
120 tx: "InstructionTranslator",
153 def var_getattr(self, tx: "InstructionTranslator", name: str) -> VariableTracker:
217 def var_getattr(self, tx: "InstructionTranslator", name: str) -> VariableTracker:
282 def var_getattr(self, tx: "InstructionTranslator", name):
388 def _setup_hook(self, tx: "InstructionTranslator", hook_method_name, args):
H A Dconstant.py102 def getitem_const(self, tx: "InstructionTranslator", arg: VariableTracker):
122 def const_getattr(self, tx: "InstructionTranslator", name):
206 def call_hasattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
233 def const_getattr(self, tx: "InstructionTranslator", name):
H A Dtensor.py220 def dynamic_getattr(self, tx: "InstructionTranslator", name):
346 def call_hasattr(self, tx: "InstructionTranslator", name):
367 def var_getattr(self, tx: "InstructionTranslator", name):
468 def unpack_var_sequence(self, tx: "InstructionTranslator", idxes=None):
1151 def create(tx: "InstructionTranslator", proxy, **options):
1161 def var_getattr(self, tx: "InstructionTranslator", name):
1310 tx: "InstructionTranslator",
H A Dbase.py238 def const_getattr(self, tx: "InstructionTranslator", name: str) -> Any:
242 def var_getattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
315 def call_hasattr(self, tx: "InstructionTranslator", name: str) -> "VariableTracker":
320 tx: "InstructionTranslator",
H A Doptimizer.py93 def var_getattr(self, tx: "InstructionTranslator", name):
289 def wrap_tensor(self, tx: "InstructionTranslator", tensor_value):
318 self, tx: "InstructionTranslator", args, kwargs, py_args, py_kwargs
H A Dsdpa.py20 def create(tx: "InstructionTranslator", value, source):
70 def var_getattr(self, tx: "InstructionTranslator", name: str) -> VariableTracker:
H A Dbuilder.py2790 def create(tx: "InstructionTranslator", value) -> VariableTracker:
2883 def passthrough(tx: "InstructionTranslator", value):
H A Diter.py41 tx: "InstructionTranslator",
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dsymbolic_convert.py300 def impl(self: "InstructionTranslator", inst: Instruction):
2632 class InstructionTranslator(InstructionTranslatorBase): class