Add learning_rates input to the SendTPUEmbeddingGradients op. This allows the
learning rate to be modified at runtime. The implementation is not yet complete. PiperOrigin-RevId: 215030536
Loading
Please sign in to comment
learning rate to be modified at runtime. The implementation is not yet complete. PiperOrigin-RevId: 215030536