Searched defs:additional_inputs (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | while_loop.py | 149 def while_loop_dense(cond_fn, body_fn, carried_inputs, additional_inputs): argument
186 def while_loop_tracing(mode, cond_fn, body_fn, carried_inputs, additional_inputs): argument
188 proxy_mode, while_loop_op, cond_fn, body_fn, carried_inputs, additional_inputs argument
230 mode, cond_fn, body_fn, carried_inputs, additional_inputs argument
237 def while_loop_func(ctx, cond_fn, body_fn, carried_inputs, additional_inputs): argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 6238 def while_loop(cond_fn, body_fn, carried_inputs, additional_inputs): argument
|
| H A D | ir.py | 7736 additional_inputs: Optional[List[TensorBox]] = None variable in WhileLoop
|