Searched refs:grads_count (Results 1 – 1 of 1) sorted by relevance
2413 auto grads_count = GetBpropGraphGradsCount(graph); in GenerateBucketSizeList() local2414 if (grads_count == 0) { in GenerateBucketSizeList()2417 return {grads_count}; in GenerateBucketSizeList()2453 auto grads_count = GetBpropGraphGradsCount(graph); in PreProcessOnSplitIndex() local2454 if (split_index_num >= grads_count) { in PreProcessOnSplitIndex()2455 …ING) << "Invalid all_reduce_fusion_config:" << *split_index << " total grads count:" << grads_count in PreProcessOnSplitIndex()2458 split_index->push_back(grads_count - 1); in PreProcessOnSplitIndex()2459 } else if (split_index_num < grads_count - 1) { in PreProcessOnSplitIndex()2460 split_index->push_back(grads_count - 1); in PreProcessOnSplitIndex()