Searched refs:unaggregated_grads (Results 1 – 1 of 1) sorted by relevance
806 auto& unaggregated_grads = gradients[id]; in ComputeGradient() local807 unaggregated_grads.push_back(in_gradients[i]); in ComputeGradient()808 if (unaggregated_grads.size() > kMinAggregateCount) { in ComputeGradient()812 size = vspace.NumElements(unaggregated_grads[0]); in ComputeGradient()817 if (unaggregated_grads.size() * size * 4 > kMinAggregateBytes) { in ComputeGradient()818 Gradient* grad = vspace.AggregateGradients(unaggregated_grads); in ComputeGradient()819 unaggregated_grads.clear(); in ComputeGradient()820 unaggregated_grads.push_back(grad); in ComputeGradient()