Tune in to the first Women in ML Symposium this Tuesday, October 19 at 9am PST Register now

Stage.Options

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.