Searched refs:all_gather_input_sizes (Results 1 – 1 of 1) sorted by relevance
176 all_gather_input_sizes: Sequence[torch.Size] = () # ND variable in ExtensionsData180 self.all_gather_input_sizes = ()488 self.all_gather_outputs, self._extensions_data.all_gather_input_sizes632 self._extensions_data.all_gather_input_sizes = [