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
)

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).

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.