Searched refs:num_on_stack (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/_dynamo/ |
| D | codegen.py | 349 def load_function_name(self, fn_name, push_null, num_on_stack=0): argument
354 if num_on_stack > 0:
357 *self.rot_n(num_on_stack + 2),
358 *self.rot_n(num_on_stack + 2),
365 *self.rot_n(num_on_stack + 1),
412 self, fn_name: str, code: types.CodeType, push_null: bool, num_on_stack=0 argument
441 output.extend(self.rot_n(num_on_stack + 2))
442 output.extend(self.rot_n(num_on_stack + 2))
445 output.extend(self.rot_n(num_on_stack + 1))
|