public final class
Stage
Stage values similar to a lightweight Enqueue.
The basic functionality of this Op is similar to a queue with many fewer capabilities and options. This Op is optimized for performance.
Nested Classes
| class | Stage.Options |
Optional attributes for
Stage
|
|
Public Methods
| static Stage.Options |
capacity
(Long capacity)
|
| static Stage.Options |
container
(String container)
|
| static Stage |
create
(
Scope
scope, Iterable<
Operand
<?>> values,
Options...
options)
Factory method to create a class wrapping a new Stage operation.
|
| static Stage.Options |
memoryLimit
(Long memoryLimit)
|
| static Stage.Options |
sharedName
(String sharedName)
|
Inherited Methods
Public Methods
public static Stage.Options capacity (Long capacity)
Parameters
| capacity | Maximum number of elements in the Staging Area. If > 0, inserts on the container will block when the capacity is reached. |
|---|
public static Stage.Options container (String container)
Parameters
| container | If non-empty, this queue is placed in the given container. Otherwise, a default container is used. |
|---|
public static Stage create ( Scope scope, Iterable< Operand <?>> values, Options... options)
Factory method to create a class wrapping a new Stage operation.
Parameters
| scope | current scope |
|---|---|
| values | a list of tensors dtypes A list of data types that inserted values should adhere to. |
| options | carries optional attributes values |
Returns
- a new instance of Stage
public static Stage.Options memoryLimit (Long memoryLimit)
Parameters
| memoryLimit | The maximum number of bytes allowed for Tensors in the Staging Area. If > 0, inserts will block until sufficient space is available. |
|---|
public static Stage.Options sharedName (String sharedName)
Parameters
| sharedName | It is necessary to match this name to the matching Unstage Op. |
|---|