@Operator public final class Stage extends PrimitiveOp
The basic functionality of this Op is similar to a queue with many fewer capabilities and options. This Op is optimized for performance.
Modifier and Type | Class and Description |
---|---|
static class |
Stage.Options
Optional attributes for
Stage |
operation
Modifier and Type | Method and Description |
---|---|
static Stage.Options |
capacity(Long capacity) |
static Stage.Options |
container(String container) |
static Stage |
create(Scope scope,
Iterable<Operand<?>> values,
Stage.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) |
equals, hashCode, op, toString
public static Stage create(Scope scope, Iterable<Operand<?>> values, Stage.Options... options)
scope
- current scopevalues
- a list of tensors
dtypes A list of data types that inserted values should adhere to.options
- carries optional attributes valuespublic static Stage.Options capacity(Long capacity)
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 memoryLimit(Long memoryLimit)
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 container(String container)
container
- If non-empty, this queue is placed in the given container. Otherwise,
a default container is used.public static Stage.Options sharedName(String sharedName)
sharedName
- It is necessary to match this name to the matching Unstage Op.Copyright © 2022. All rights reserved.