Searched refs:new_gradients (Results 1 – 2 of 2) sorted by relevance
542 Gradient* new_gradients = nullptr; in ComputeGradient() local544 new_gradients = grad_it->second.at(0); in ComputeGradient()546 new_gradients = vspace.AggregateGradients(grad_it->second); in ComputeGradient()552 grad_it->second.push_back(new_gradients); in ComputeGradient()553 vspace.MarkAsResult(new_gradients); in ComputeGradient()555 out_gradients.push_back(new_gradients); in ComputeGradient()
305 std::vector<float> new_gradients(gradients_shape.num_elements()); in AddToTensorAccumulator() local307 new_gradients[j] = gradients(i, j); in AddToTensorAccumulator()313 (*stats_map)[key] = {new_gradients, new_hessians}; in AddToTensorAccumulator()