Searched refs:unaggregated_grads (Results 1 – 1 of 1) sorted by relevance
585 auto& unaggregated_grads = gradients[id]; in ComputeGradient() local586 unaggregated_grads.push_back(in_gradients[i]); in ComputeGradient()587 if (unaggregated_grads.size() > kMinAggregateCount) { in ComputeGradient()591 size = vspace.NumElements(unaggregated_grads[0]); in ComputeGradient()596 if (unaggregated_grads.size() * size * 4 > kMinAggregateBytes) { in ComputeGradient()597 Gradient* grad = vspace.AggregateGradients(unaggregated_grads); in ComputeGradient()598 unaggregated_grads.clear(); in ComputeGradient()599 unaggregated_grads.push_back(grad); in ComputeGradient()