Searched refs:enqueue_tpu_embedding_arbitrary_tensor_batch (Results 1 – 3 of 3) sorted by relevance
549 def enqueue_tpu_embedding_arbitrary_tensor_batch(sample_indices_or_row_splits, function597 return gen_tpu_ops.enqueue_tpu_embedding_arbitrary_tensor_batch(607 enqueue_tpu_embedding_arbitrary_tensor_batch.__doc__ = (608 gen_tpu_ops.enqueue_tpu_embedding_arbitrary_tensor_batch.__doc__)
996 return tpu_ops.enqueue_tpu_embedding_arbitrary_tensor_batch(
1849 return tpu_ops.enqueue_tpu_embedding_arbitrary_tensor_batch(