tf.raw_ops.GetMinibatchesInCsrWithPhysicalReplica

View aliases

Compat aliases for migration

tf.compat.v1.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`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/raw_ops/GetMinibatchesInCsrWithPhysicalReplica