Searched refs:grad_it (Results 1 – 2 of 2) sorted by relevance
747 auto grad_it = gradients.find(id); in ComputeGradient() local748 if (grad_it == gradients.end()) { in ComputeGradient()761 if (grad_it->second.size() == 1) { in ComputeGradient()762 new_gradients = grad_it->second.at(0); in ComputeGradient()764 new_gradients = vspace.AggregateGradients(grad_it->second); in ComputeGradient()766 if (sources_set.find(grad_it->first) == sources_set.end()) { in ComputeGradient()767 gradients.erase(grad_it); in ComputeGradient()769 grad_it->second.clear(); in ComputeGradient()770 grad_it->second.push_back(new_gradients); in ComputeGradient()838 auto grad_it = gradients.find(id); in ComputeGradient() local[all …]
537 auto grad_it = grads.find(argnum); in ChangeFunctionInputSignature() local538 if (grad_it == grads.end()) continue; in ChangeFunctionInputSignature()540 for (const string& source : grad_it->getSecond()) { in ChangeFunctionInputSignature()606 auto grad_it = grads.find(i); in HandleWhileOp() local608 if (grad_it == grads.end()) continue; in HandleWhileOp()609 for (const string& source : grad_it->getSecond()) { in HandleWhileOp()673 auto grad_it = grads.find(i); in HandleIfOp() local675 if (grad_it == grads.end()) continue; in HandleIfOp()676 for (const string& source : grad_it->getSecond()) { in HandleIfOp()737 auto grad_it = it->getSecond().grads.find(source); in HandlePartitionedCallOp() local[all …]