Searched refs:getUnmarkedParamIndicesForIteration (Results 1 – 2 of 2) sorted by relevance
551 std::vector<size_t> getUnmarkedParamIndicesForIteration();
1494 std::vector<size_t> Reducer::getUnmarkedParamIndicesForIteration() { in getUnmarkedParamIndicesForIteration() function in c10d::Reducer1918 auto unmarked_param_indices = getUnmarkedParamIndicesForIteration(); in ensure_prior_reduction_finished()