Fix bug in tf.gradients.
This is fixing in a bug where we'd incorrectly match up gradients tensors with the corresponding forward-pass tensors due to filtering out captured EagerTensors. This could lead to errors about incompatible gradient shapes or, even worse, the wrong gradient value. PiperOrigin-RevId: 235271326
Loading
Please sign in to comment