Stage

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  

Constants

String OP_NAME The name of this op, as known by TensorFlow core engine

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

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "Stage"

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.