Remove identity ops for ys added during gradient computation. This was added...

Remove identity ops for ys added during gradient computation. This was added to avoid issues with computing gradients when ys were dependent.
The real issue behind that has however since been fixed so adding identity ops is no longer relevant.

PiperOrigin-RevId: 207974344