Searched defs:tensors_to_clone (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | triton_kernel_wrap.py | 545 def __call__(self, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone): argument
667 *, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone argument
688 mode, *, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone argument
704 mode, *, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone argument
721 ctx, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/ |
| H A D | reinplace.py | 501 tensors_to_clone, argument
|