tf.raw_ops.GetMinibatchesInCsrWithPhysicalReplica
tf.raw_ops.GetMinibatchesInCsrWithPhysicalReplica(
program_key,
row_ids,
col_ids,
gains,
splits,
id_counts,
sample_count,
num_replica,
max_minibatches_per_sc,
max_ids_per_chip_per_sample,
table_vocab_size,
feature_width,
num_sc_per_chip,
table_name,
mini_batch_in_csr,
name=None
)
| Args |
|---|
program_key | A Tensor of type string. |
row_ids | A Tensor of type int32. |
col_ids | A Tensor of type int32. |
gains | A Tensor of type float32. |
splits | A Tensor of type int64. |
id_counts | A Tensor of type int32. |
sample_count | An int that is >= 1. |
num_replica | An int that is >= 1. |
max_minibatches_per_sc | An int that is >= 1. |
max_ids_per_chip_per_sample | An int that is >= 1. |
table_vocab_size | An int that is >= 1. |
feature_width | An int that is >= 1. |
num_sc_per_chip | An int that is >= 1. |
table_name | A string. |
mini_batch_in_csr | A string. |
name | A name for the operation (optional). |
| Returns |
|---|
A tuple of Tensor objects (row_pointers, sorted_sample_ids, sorted_token_ids, sorted_gains, row_pointers_unpadded_size, ids_unpadded_size, num_minibatches_per_physical_sparse_core). |
row_pointers | A Tensor of type int32. |
sorted_sample_ids | A Tensor of type int32. |
sorted_token_ids | A Tensor of type int32. |
sorted_gains | A Tensor of type float32. |
row_pointers_unpadded_size | A Tensor of type int32. |
ids_unpadded_size | A Tensor of type int32. |
num_minibatches_per_physical_sparse_core | A Tensor of type int32. |