Searched refs:send_tpu_embedding_gradients (Results 1 – 2 of 2) sorted by relevance
245 def send_tpu_embedding_gradients(inputs, function273 return gen_tpu_ops.send_tpu_embedding_gradients(277 send_tpu_embedding_gradients.__doc__ = (278 gen_tpu_ops.send_tpu_embedding_gradients.__doc__)
738 return tpu_ops.send_tpu_embedding_gradients(