Move sorting of variables from the implicit_grad family of functions
up to GradientTape.watched_variables() so we also get deterministic behavior when using the GradientTape as is done in Optimizer.compute_gradients(). PiperOrigin-RevId: 186533323
Loading
Please sign in to comment