Searched defs:output_split_sizes (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | Functional.cpp | 259 std::vector<int64_t> output_split_sizes, in all_to_all_single()
403 std::vector<int64_t> output_split_sizes, in forward()
422 const std::vector<int64_t>& output_split_sizes = in backward() local
450 const std::vector<int64_t>& output_split_sizes, in all_to_all_single_autograd()
|
| /aosp_15_r20/external/pytorch/torch/distributed/nn/ |
| H A D | functional.py | 177 output_split_sizes=None, argument
411 def forward(ctx, group, output, output_split_sizes, input_split_sizes, input): argument
|
| /aosp_15_r20/external/pytorch/torch/distributed/ |
| H A D | _functional_collectives.py | 886 input, output_split_sizes, input_split_sizes, *args, **kwargs argument
1074 output_split_sizes=None, argument
|
| H A D | distributed_c10d.py | 3883 output_split_sizes=None, argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | lowering.py | 6412 def _all_to_all_single(inp, output_split_sizes, input_split_sizes, group_name): argument
|