public static class
Stage.Options
Optional attributes for
Stage
Public Methods
Stage.Options |
capacity
(Long capacity)
|
Stage.Options |
container
(String container)
|
Stage.Options |
memoryLimit
(Long memoryLimit)
|
Stage.Options |
sharedName
(String sharedName)
|
Inherited Methods
Public Methods
public 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 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 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 Stage.Options sharedName (String sharedName)
Parameters
sharedName | It is necessary to match this name to the matching Unstage Op. |
---|