Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dtape.h747 auto grad_it = gradients.find(id); in ComputeGradient() local
838 auto grad_it = gradients.find(id); in ComputeGradient() local
874 auto grad_it = gradients.find(tensor_id); in ComputeGradient() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_array_ops_decomposition.cc537 auto grad_it = grads.find(argnum); in ChangeFunctionInputSignature() local
606 auto grad_it = grads.find(i); in HandleWhileOp() local
673 auto grad_it = grads.find(i); in HandleIfOp() local
737 auto grad_it = it->getSecond().grads.find(source); in HandlePartitionedCallOp() local