Searched defs:use_local_output (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/parallel/ |
| H A D | style.py | 121 def _prepare_output_fn(output_layouts, use_local_output, mod, outputs, device_mesh): argument
221 def _prepare_output_fn(output_layouts, use_local_output, mod, outputs, device_mesh): argument
320 def _prepare_output_fn(use_local_output, mod, outputs, device_mesh): argument
|