tf.raw_ops.BarrierTakeMany

Takes the given number of completed elements from a barrier.

tf.raw_ops.BarrierTakeMany(
    handle,
    num_elements,
    component_types,
    allow_small_batch=False,
    wait_for_incomplete=False,
    timeout_ms=-1,
    name=None
)

This operation concatenates completed-element component tensors along the 0th dimension to make a single component tensor.

Elements come out of the barrier when they are complete, and in the order in which they were placed into the barrier. The indices output provides information about the batch in which each element was originally inserted into the barrier.

handleA Tensor of type mutable string. The handle to a barrier.
num_elementsA Tensor of type int32. A single-element tensor containing the number of elements to take.
component_typesA list of tf.DTypes that has length >= 1. The type of each component in a value.
allow_small_batchAn optional bool. Defaults to False. Allow to return less than num_elements items if barrier is already closed.
wait_for_incompleteAn optional bool. Defaults to False.
timeout_msAn optional int. Defaults to -1. If the queue is empty, this operation will block for up to timeout_ms milliseconds. Note: This option is not supported yet.
nameA name for the operation (optional).

A tuple of Tensor objects (indices, keys, values).
indicesA Tensor of type int64.
keysA Tensor of type string.
valuesA list of Tensor objects of type component_types.