Searched refs:send_tpu_embedding_gradients (Results 1 – 4 of 4) sorted by relevance
248 def send_tpu_embedding_gradients(inputs, function276 return gen_tpu_ops.send_tpu_embedding_gradients(280 send_tpu_embedding_gradients.__doc__ = (281 gen_tpu_ops.send_tpu_embedding_gradients.__doc__)
15 // CHECK-LABEL: func @send_tpu_embedding_gradients16 func @send_tpu_embedding_gradients(%arg0: tensor<512x256xf32>) -> () {
662 op = tpu_ops.send_tpu_embedding_gradients(
1916 return tpu_ops.send_tpu_embedding_gradients(