#include <data_flow_ops.h>
Takes the given number of completed elements from a barrier.
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:
Optional attributes (see Attrs
):
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 | |
operation | |
values |
Public static functions | |
---|---|
AllowSmallBatch(bool x) | |
TimeoutMs(int64 x) | |
WaitForIncomplete(bool x) |
Structs | |
---|---|
tensorflow::ops::BarrierTakeMany::Attrs | Optional attribute setters for BarrierTakeMany. |
::tensorflow::Output indices
::tensorflow::Output keys
Operation operation
::tensorflow::OutputList values
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 )
Attrs AllowSmallBatch( bool x )
Attrs TimeoutMs( int64 x )
Attrs WaitForIncomplete( bool x )
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/cc/class/tensorflow/ops/barrier-take-many