Commit ca43fe82 authored by Ali Yahya's avatar Ali Yahya Committed by TensorFlower Gardener
Browse files

TFE: Improves the interfaces of tape.watch_variable() and implicit_grad(). 

tape.watch_variable() replaces tape.watch() and now is called on ResourceVariable objects instead of their underlying handles.

implicit_grad() now returns a list of (gradient, variable) pairs to be consistent with tf.Optimizer's interface.

PiperOrigin-RevId: 168232055
parent b72862df
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment