Searched defs:grad_it (Results 1 – 2 of 2) sorted by relevance
747 auto grad_it = gradients.find(id); in ComputeGradient() local838 auto grad_it = gradients.find(id); in ComputeGradient() local874 auto grad_it = gradients.find(tensor_id); in ComputeGradient() local
537 auto grad_it = grads.find(argnum); in ChangeFunctionInputSignature() local606 auto grad_it = grads.find(i); in HandleWhileOp() local673 auto grad_it = grads.find(i); in HandleIfOp() local737 auto grad_it = it->getSecond().grads.find(source); in HandlePartitionedCallOp() local