Searched defs:_calculate_single_rank_operations (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/pipelining/ | ||
| H A D | schedules.py | 1645 def _calculate_single_rank_operations(self, rank): member in ScheduleLoopedBFS |