org.tensorflow.op.core

Class MapStage.Options

java.lang.Object

org.tensorflow.op.core.MapStage.Options

Enclosing class:

MapStage

public static class MapStage.Options
extends Object

Optional attributes for MapStage

Method Summary

Modifier and Type	Method and Description
MapStage.Options	capacity(Long capacity)
MapStage.Options	container(String container)
MapStage.Options	memoryLimit(Long memoryLimit)
MapStage.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 MapStage.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 MapStage.Options memoryLimit(Long memoryLimit)

Parameters:

memoryLimit -

container

public MapStage.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 MapStage.Options sharedName(String sharedName)

Parameters:

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