Home
last modified time | relevance | path

Searched refs:create_call_function (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dcodegen.py15 create_call_function,
170 output.extend(create_call_function(2, False))
190 output.extend(create_call_function(0, False))
207 output.extend(create_call_function(1, False))
215 output.extend(create_call_function(0, False))
395 *create_call_function(0, False),
403 self.extend_output(create_call_function(nargs, push_null=push_null))
477 self.extend_output(create_call_function(1, False))
481 self.extend_output(create_call_function(len(graphargs), False))
488 output = create_call_function(nargs, push_null)
[all …]
H A Dsource.py10 from .bytecode_transformation import create_call_function, create_instruction
166 codegen.extend_output(create_call_function(0, False))
179 codegen.extend_output(create_call_function(0, False))
306 create_call_function(1 if self.idx is not None else 0, False)
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))
625 codegen.extend_output(create_call_function(1, False))
659 codegen.extend_output(create_call_function(1, False))
H A Dresume_execution.py9 create_call_function,
89 *create_call_function(len(load_args), False),
272 *create_call_function(2, False),
299 *create_call_function(len(load_args), False),
347 insts.extend(create_call_function(len(tup), False))
H A Dside_effects.py13 create_call_function,
415 cg.extend_output(create_call_function(0, False))
426 cg.extend_output(create_call_function(1, False))
514 cg.extend_output(create_call_function(1, False))
680 *create_call_function(1, False), # setstate
H A Doutput_graph.py37 create_call_function,
1088 random_calls_instructions.extend(create_call_function(0, False))
1182 cg.extend_output(create_call_function(len(args), False))
H A Dsymbolic_convert.py51 create_call_function,
674 create_call_function(inst.arg, False)
2946 cg.extend_output(create_call_function(nargs, False))
H A Dbytecode_transformation.py264 def create_call_function(nargs, push_null) -> List[Instruction]: function
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dlists.py15 from ..bytecode_transformation import create_call_function, create_instruction
305 codegen.extend_output(create_call_function(3, False))
473 *create_call_function(1, False),
641 ] + create_call_function(1, False)
742 + create_call_function(1, False)
982 codegen.extend_output(create_call_function(1, False))
H A Dsdpa.py6 from ..bytecode_transformation import create_call_function
65 codegen.extend_output(create_call_function(len(self.param_vars), False))
H A Dfunctions.py14 from ..bytecode_transformation import create_call_function, create_rot_n
557 codegen.extend_output(create_call_function(7, push_null=True))
562 codegen.extend_output(create_call_function(1, True))
564 codegen.extend_output(create_call_function(1, True))
779 codegen.extend_output(create_call_function(len(self.args) + 1, True))
H A Diter.py9 from ..bytecode_transformation import create_call_function, create_instruction
242 codegen.extend_output(create_call_function(1, False))
273 codegen.extend_output(create_call_function(2, False))
H A Ddicts.py13 from ..bytecode_transformation import create_call_function, create_instruction
212 *create_call_function(1, False),
577 codegen.extend_output(create_call_function(0, False))
855 codegen.extend_output(create_call_function(1, False))
H A Dctx_manager.py13 create_call_function,
103 codegen.extend_output(create_call_function(len(target_values), False))
1142 create_call_function(len(self.ctx.target_values), False)
H A Dmisc.py18 from ..bytecode_transformation import create_call_function, create_instruction
79 codegen.extend_output(create_call_function(2, False))
81 codegen.extend_output(create_call_function(1, False))
468 codegen.extend_output(create_call_function(1, False))
H A Duser_defined.py22 from ..bytecode_transformation import create_call_function
1372 codegen.extend_output(create_call_function(0, False))