Searched refs:grads_to_keep (Results 1 – 1 of 1) sorted by relevance
352 std::vector<Output> grads_to_keep; in SumGradients() local355 grads_to_keep.push_back(o); in SumGradients()358 if (grads_to_keep.empty()) { in SumGradients()361 } else if (grads_to_keep.size() == 1) { in SumGradients()363 *grad = grads_to_keep[0]; in SumGradients()367 *grad = ops::AddN(scope_, grads_to_keep); in SumGradients()