Searched defs:carried_inputs (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/_higher_order_ops/ |
| D | while_loop.py | 62 def while_loop(cond_fn, body_fn, carried_inputs): argument
123 def _validate_input(cond_fn, body_fn, carried_inputs): argument
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
|
| /external/pytorch/torch/_inductor/ |
| D | lowering.py | 6238 def while_loop(cond_fn, body_fn, carried_inputs, additional_inputs): argument
|
| D | ir.py | 7735 carried_inputs: Optional[List[TensorBox]] = None variable in WhileLoop
|