ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Unstage

public final class Unstage

Op is similar to a lightweight Dequeue.

The basic functionality is similar to dequeue with many fewer capabilities and options. This Op is optimized for performance.

Nested Classes

class Unstage.Options Optional attributes for Unstage  

Public Methods

static Unstage.Options
capacity(Long capacity)
static Unstage.Options
container(String container)
static Unstage
create(Scope scope, List<Class<?>> dtypes, Options... options)
Factory method to create a class wrapping a new Unstage operation.
Iterator<Operand<Object>>
static Unstage.Options
memoryLimit(Long memoryLimit)
static Unstage.Options
sharedName(String sharedName)
List<Output<?>>
values()

Inherited Methods

Public Methods

public static Unstage.Options capacity (Long capacity)

public static Unstage.Options container (String container)

public static Unstage create (Scope scope, List<Class<?>> dtypes, Options... options)

Factory method to create a class wrapping a new Unstage operation.

Parameters
scope current scope
options carries optional attributes values
Returns
  • a new instance of Unstage

public Iterator<Operand<Object>> iterator ()

public static Unstage.Options memoryLimit (Long memoryLimit)

public static Unstage.Options sharedName (String sharedName)

public List<Output<?>> values ()