Register gradient for TPUEmbeddingActivation in one place.
Factor dummy variable out of tpu_embedding.py. PiperOrigin-RevId: 234178740
Loading
Please sign in to comment
Factor dummy variable out of tpu_embedding.py. PiperOrigin-RevId: 234178740