Removes assertion which is not always true.
It's perfectly fine to have float tensors which are loop variables which are not related to the gradient objective. PiperOrigin-RevId: 226235207
Loading
Please sign in to comment
It's perfectly fine to have float tensors which are loop variables which are not related to the gradient objective. PiperOrigin-RevId: 226235207