org.tensorflow.op.core

Class MapStage

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.MapStage

All Implemented Interfaces:

Op

@Operator
public final class MapStage
extends PrimitiveOp

Stage (key, values) in the underlying container which behaves like a hashtable.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	MapStage.Options Optional attributes for MapStage

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
static MapStage.Options	capacity(Long capacity)
static MapStage.Options	container(String container)
static MapStage	create(Scope scope, Operand<Long> key, Operand<Integer> indices, Iterable<Operand<?>> values, List<Class<?>> dtypes, MapStage.Options... options) Factory method to create a class to wrap a new MapStage operation to the graph.
static MapStage.Options	memoryLimit(Long memoryLimit)
static MapStage.Options	sharedName(String sharedName)

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Method Detail

create

public static MapStage create(Scope scope,
                              Operand<Long> key,
                              Operand<Integer> indices,
                              Iterable<Operand<?>> values,
                              List<Class<?>> dtypes,
                              MapStage.Options... options)

Factory method to create a class to wrap a new MapStage operation to the graph.

Parameters:

scope - current graph scope

key - int64

indices -

values - a list of tensors dtypes A list of data types that inserted values should adhere to.

dtypes -

options - carries optional attributes values

Returns:

a new instance of MapStage

capacity

public static 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 static MapStage.Options memoryLimit(Long memoryLimit)

Parameters:

memoryLimit -

container

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

Parameters:

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