Home
last modified time | relevance | path

Searched refs:extend_output (Results 1 – 14 of 14) sorted by relevance

/external/pytorch/torch/_dynamo/
Dsource.py166 codegen.extend_output(create_call_function(0, False))
179 codegen.extend_output(create_call_function(0, False))
203 codegen.extend_output(codegen.create_load_attrs(self.member))
224 codegen.extend_output(codegen.create_load_attrs(self.member))
305 codegen.extend_output(
422 codegen.extend_output(codegen.create_load_attrs(self.field))
488 codegen.extend_output(create_call_function(2, False))
503 codegen.extend_output(create_call_function(2, False))
517 codegen.extend_output(create_call_function(1, False))
540 codegen.extend_output(create_call_function(2, False))
[all …]
Dside_effects.py415 cg.extend_output(create_call_function(0, False))
426 cg.extend_output(create_call_function(1, False))
440 cg.extend_output(
510 cg.extend_output([cg.create_load_attr(name)])
514 cg.extend_output(create_call_function(1, False))
536 cg.extend_output(
551 cg.extend_output(
556 cg.extend_output(
689 cg.extend_output(suffix)
Dcodegen.py283 def extend_output(self, insts): member in PyCodegen
328 self.extend_output(create_call_method(nargs))
403 self.extend_output(create_call_function(nargs, push_null=push_null))
463 self.extend_output(self.load_function_name(fn_name, True))
469 lambda: self.extend_output(
477 self.extend_output(create_call_function(1, False))
481 self.extend_output(create_call_function(len(graphargs), False))
Dsymbolic_convert.py657 cg.extend_output(b.resume_fn().try_except(cg.code_options, cleanup))
2866 cg.extend_output(create_swap(len(self.stack) - i + 1))
2900 cg.extend_output(cg.pop_null())
2943 cg.extend_output(cg.load_function_name(name, True, stack_len))
2945 cg.extend_output([cg.create_load(k) for k in argnames])
2946 cg.extend_output(create_call_function(nargs, False))
Doutput_graph.py1182 cg.extend_output(create_call_function(len(args), False))
1183 cg.extend_output([create_instruction("POP_TOP")])
/external/pytorch/torch/_dynamo/variables/
Dfunctions.py530 codegen.extend_output([codegen._create_load_const(self.f_globals)])
536 codegen.extend_output([codegen.create_load_const(None)])
541 codegen.extend_output([codegen.create_load_const(None)])
546 codegen.extend_output([codegen.create_load_const(None)])
551 codegen.extend_output([codegen._create_load_const(annotations)])
555 codegen.extend_output([codegen.create_load_const(None)])
557 codegen.extend_output(create_call_function(7, push_null=True))
562 codegen.extend_output(create_call_function(1, True))
563 codegen.extend_output(create_rot_n(2))
564 codegen.extend_output(create_call_function(1, True))
[all …]
Diter.py234 lambda: codegen.extend_output(
242 codegen.extend_output(create_call_function(1, False))
264 lambda: codegen.extend_output(
273 codegen.extend_output(create_call_function(2, False))
428 codegen.extend_output(
470 codegen.extend_output(
Ddicts.py196 lambda: codegen.extend_output(
209 codegen.extend_output(
571 lambda: codegen.extend_output(
577 codegen.extend_output(create_call_function(0, False))
851 codegen.extend_output([codegen._create_load_const(self.user_cls)])
855 codegen.extend_output(create_call_function(1, False))
863 codegen.extend_output(codegen.create_call_function_kw(len(keys), keys, False))
981 lambda: codegen.extend_output(
Dlists.py305 codegen.extend_output(create_call_function(3, False))
470 codegen.extend_output(
642 codegen.extend_output(build_torch_size)
738 codegen.extend_output(
877 codegen.extend_output(
982 codegen.extend_output(create_call_function(1, False))
Dsdpa.py65 codegen.extend_output(create_call_function(len(self.param_vars), False))
Dmisc.py79 codegen.extend_output(create_call_function(2, False))
81 codegen.extend_output(create_call_function(1, False))
460 lambda: codegen.extend_output(
468 codegen.extend_output(create_call_function(1, False))
616 codegen.extend_output(
1016 codegen.extend_output(codegen.create_load_attrs(self.name))
1376 lambda: codegen.extend_output(
1709 lambda: codegen.extend_output(
Dctx_manager.py102 codegen.extend_output([codegen.create_load_const(val) for val in target_values])
103 codegen.extend_output(create_call_function(len(target_values), False))
411 cg.extend_output(cg.create_call_function_kw(len(keys), keys, False))
1138 codegen.extend_output(
1141 codegen.extend_output(
Dtorch.py170 codegen.extend_output(
Duser_defined.py1372 codegen.extend_output(create_call_function(0, False))