@Operator public final class MapStage extends PrimitiveOp
Modifier and Type | Class and Description |
---|---|
static class |
MapStage.Options
Optional attributes for
MapStage |
operation
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 wrapping a new MapStage operation.
|
static MapStage.Options |
memoryLimit(Long memoryLimit) |
static MapStage.Options |
sharedName(String sharedName) |
equals, hashCode, op, toString
public static MapStage create(Scope scope, Operand<Long> key, Operand<Integer> indices, Iterable<Operand<?>> values, List<Class<?>> dtypes, MapStage.Options... options)
scope
- current scopekey
- int64indices
- values
- a list of tensors
dtypes A list of data types that inserted values should adhere to.dtypes
- options
- carries optional attributes valuespublic static MapStage.Options capacity(Long capacity)
capacity
- Maximum number of elements in the Staging Area. If > 0, inserts
on the container will block when the capacity is reached.public static MapStage.Options memoryLimit(Long memoryLimit)
memoryLimit
- public static MapStage.Options container(String container)
container
- If non-empty, this queue is placed in the given container. Otherwise,
a default container is used.public static MapStage.Options sharedName(String sharedName)
sharedName
- It is necessary to match this name to the matching Unstage Op.Copyright © 2022. All rights reserved.