Searched defs:grad_it (Results 1 – 3 of 3) sorted by relevance
536 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
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
383 auto grad_it = gradient_map.find(signature.name()); in ConvertFunctionAttributes() local