Searched refs:enqueue_tpu_embedding_ragged_tensor_batch (Results 1 – 3 of 3) sorted by relevance
474 def enqueue_tpu_embedding_ragged_tensor_batch(sample_splits, function536 return gen_tpu_ops.enqueue_tpu_embedding_ragged_tensor_batch(549 enqueue_tpu_embedding_ragged_tensor_batch.__doc__ = (550 gen_tpu_ops.enqueue_tpu_embedding_ragged_tensor_batch.__doc__)
1018 return tpu_ops.enqueue_tpu_embedding_ragged_tensor_batch(
1754 return tpu_ops.enqueue_tpu_embedding_ragged_tensor_batch(