Searched refs:enqueue_tpu_embedding_ragged_tensor_batch (Results 1 – 3 of 3) sorted by relevance
457 def enqueue_tpu_embedding_ragged_tensor_batch(sample_splits, function519 return gen_tpu_ops.enqueue_tpu_embedding_ragged_tensor_batch(532 enqueue_tpu_embedding_ragged_tensor_batch.__doc__ = (533 gen_tpu_ops.enqueue_tpu_embedding_ragged_tensor_batch.__doc__)
1017 return tpu_ops.enqueue_tpu_embedding_ragged_tensor_batch(
1754 return tpu_ops.enqueue_tpu_embedding_ragged_tensor_batch(