tensorflow::ops::BarrierTakeMany

#include <data_flow_ops.h>

Takes the given number of completed elements from a barrier.

Summary

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.

Arguments:

  • scope: A Scope object
  • handle: The handle to a barrier.
  • num_elements: A single-element tensor containing the number of elements to take.
  • component_types: The type of each component in a value.

Optional attributes (see Attrs):

  • allow_small_batch: Allow to return less than num_elements items if barrier is already closed.
  • timeout_ms: If the queue is empty, this operation will block for up to timeout_ms milliseconds. Note: This option is not supported yet.

Returns:

  • Output indices: A one-dimensional tensor of indices, with length num_elems. These indices refer to the batch in which the values were placed into the barrier (starting with MIN_LONG and increasing with each BarrierInsertMany).
  • Output keys: A one-dimensional tensor of keys, with length num_elements.
  • OutputList values: One any-dimensional tensor per component in a barrier element. All values have length num_elements in the 0th dimension.

Constructors and Destructors

BarrierTakeMany(const ::tensorflow::Scope & scope, ::tensorflow::Input handle, ::tensorflow::Input num_elements, const DataTypeSlice & component_types)
BarrierTakeMany(const ::tensorflow::Scope & scope, ::tensorflow::Input handle, ::tensorflow::Input num_elements, const DataTypeSlice & component_types, const BarrierTakeMany::Attrs & attrs)

Public attributes

indices
keys
values

Public static functions

AllowSmallBatch(bool x)
TimeoutMs(int64 x)
WaitForIncomplete(bool x)

Structs

tensorflow::ops::BarrierTakeMany::Attrs

Optional attribute setters for BarrierTakeMany.

Public attributes

indices

::tensorflow::Output indices

keys

::tensorflow::Output keys

values

::tensorflow::OutputList values

Public functions

BarrierTakeMany

 BarrierTakeMany(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_elements,
  const DataTypeSlice & component_types
)

BarrierTakeMany

 BarrierTakeMany(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input handle,
  ::tensorflow::Input num_elements,
  const DataTypeSlice & component_types,
  const BarrierTakeMany::Attrs & attrs
)

Public static functions

AllowSmallBatch

Attrs AllowSmallBatch(
  bool x
)

TimeoutMs

Attrs TimeoutMs(
  int64 x
)

WaitForIncomplete

Attrs WaitForIncomplete(
  bool x
)