Home
last modified time | relevance | path

Searched refs:grad_it (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dtape.h530 auto grad_it = gradients.find(id); in ComputeGradient() local
531 if (grad_it == gradients.end()) { in ComputeGradient()
543 if (grad_it->second.size() == 1) { in ComputeGradient()
544 new_gradients = grad_it->second.at(0); in ComputeGradient()
546 new_gradients = vspace.AggregateGradients(grad_it->second); in ComputeGradient()
548 if (sources_set.find(grad_it->first) == sources_set.end()) { in ComputeGradient()
549 gradients.erase(grad_it); in ComputeGradient()
551 grad_it->second.clear(); in ComputeGradient()
552 grad_it->second.push_back(new_gradients); in ComputeGradient()
617 auto grad_it = gradients.find(id); in ComputeGradient() local
[all …]