Searched refs:enqueue_tpu_embedding_sparse_tensor_batch (Results 1 – 3 of 3) sorted by relevance
377 def enqueue_tpu_embedding_sparse_tensor_batch(sample_indices, function439 return gen_tpu_ops.enqueue_tpu_embedding_sparse_tensor_batch(452 enqueue_tpu_embedding_sparse_tensor_batch.__doc__ = (453 gen_tpu_ops.enqueue_tpu_embedding_sparse_tensor_batch.__doc__)
1026 return tpu_ops.enqueue_tpu_embedding_sparse_tensor_batch(
1760 return tpu_ops.enqueue_tpu_embedding_sparse_tensor_batch(