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.

Args:

  • handle: A Tensor of type mutable string. The handle to a barrier.
  • num_elements: A Tensor of type int32. A single-element tensor containing the number of elements to take.
  • component_types: A list of tf.DTypes that has length >= 1. The type of each component in a value.
  • allow_small_batch: An optional bool. Defaults to False. Allow to return less than num_elements items if barrier is already closed.
  • wait_for_incomplete: An optional bool. Defaults to False.
  • timeout_ms: An 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.
  • name: A name for the operation (optional).

Returns:

A tuple of Tensor objects (indices, keys, values).

  • indices: A Tensor of type int64.
  • keys: A Tensor of type string.
  • values: A list of Tensor objects of type component_types.