@Operator public final class Barrier extends PrimitiveOp implements Operand<String>
A barrier represents a key-value map, where each key is a string, and each value is a tuple of tensors.
At runtime, the barrier contains 'complete' and 'incomplete' elements. A complete element has defined tensors for all components of its value tuple, and may be accessed using BarrierTakeMany. An incomplete element has some undefined components in its value tuple, and may be updated using BarrierInsertMany.
Modifier and Type | Class and Description |
---|---|
static class |
Barrier.Options
Optional attributes for
Barrier |
operation
Modifier and Type | Method and Description |
---|---|
Output<String> |
asOutput()
Returns the symbolic handle of a tensor.
|
static Barrier.Options |
capacity(Long capacity) |
static Barrier.Options |
container(String container) |
static Barrier |
create(Scope scope,
List<Class<?>> componentTypes,
Barrier.Options... options)
Factory method to create a class wrapping a new Barrier operation.
|
Output<String> |
handle()
The handle to the barrier.
|
static Barrier.Options |
shapes(List<Shape> shapes) |
static Barrier.Options |
sharedName(String sharedName) |
equals, hashCode, op, toString
public static Barrier create(Scope scope, List<Class<?>> componentTypes, Barrier.Options... options)
scope
- current scopecomponentTypes
- The type of each component in a value.options
- carries optional attributes valuespublic static Barrier.Options shapes(List<Shape> shapes)
shapes
- The shape of each component in a value. Each shape must be 1 in the
first dimension. The length of this attr must be the same as the length of
component_types.public static Barrier.Options capacity(Long capacity)
capacity
- The capacity of the barrier. The default capacity is MAX_INT32,
which is the largest capacity of the underlying queue.public static Barrier.Options container(String container)
container
- If non-empty, this barrier is placed in the given container.
Otherwise, a default container is used.public static Barrier.Options sharedName(String sharedName)
sharedName
- If non-empty, this barrier will be shared under the given name
across multiple sessions.public Output<String> asOutput()
Operand
Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.
asOutput
in interface Operand<String>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.