tf.gradients?

Maybe this is the most important question.
In the issue @Bhack linked to, Allen mentioned tf.gratient’s colocate_gradient_with_op option.
But that doesn’t exist in tf.gradients, only tf.compat.v1.gradients.
What’s going on?
Compare the code for tf.gradients to tf.compat.v1.gradients 1.
colocate_gradients_with_ops is not an option in tf2 because it is always set to True.
So maybe if you just use one big tf.gradients instead of tf.GradientTape,  this issue just dissapears.