Searched defs:input_split_sizes (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | Functional.cpp | 260 std::vector<int64_t> input_split_sizes, in all_to_all_single()
405 std::vector<int64_t> input_split_sizes, in forward()
424 const std::vector<int64_t>& input_split_sizes = in backward() local
451 const std::vector<int64_t>& input_split_sizes, in all_to_all_single_autograd()
|
| /aosp_15_r20/external/pytorch/torch/distributed/nn/ |
| H A D | functional.py | 178 input_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
1075 input_split_sizes=None, argument
|
| H A D | distributed_c10d.py | 3884 input_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
|