org.tensorflow.op.core

Class Stage.Options

java.lang.Object

org.tensorflow.op.core.Stage.Options

Enclosing class:

Stage

public static class Stage.Options
extends Object

Optional attributes for Stage

Method Summary

Modifier and Type	Method and Description
Stage.Options	capacity(Long capacity)
Stage.Options	container(String container)
Stage.Options	memoryLimit(Long memoryLimit)
Stage.Options	sharedName(String sharedName)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

capacity

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.

memoryLimit

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.

container

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.

sharedName

public Stage.Options sharedName(String sharedName)

Parameters:

sharedName - It is necessary to match this name to the matching Unstage Op.