| Package | Description |
|---|---|
| org.tensorflow.op | |
| org.tensorflow.op.core |
| Modifier and Type | Method and Description |
|---|---|
StageSize |
Ops.stageSize(List<Class<?>> dtypes,
StageSize.Options... options)
Adds an
StageSize operation to the graph |
| Modifier and Type | Method and Description |
|---|---|
static StageSize.Options |
StageSize.capacity(Long capacity) |
StageSize.Options |
StageSize.Options.capacity(Long capacity) |
static StageSize.Options |
StageSize.container(String container) |
StageSize.Options |
StageSize.Options.container(String container) |
static StageSize.Options |
StageSize.memoryLimit(Long memoryLimit) |
StageSize.Options |
StageSize.Options.memoryLimit(Long memoryLimit) |
static StageSize.Options |
StageSize.sharedName(String sharedName) |
StageSize.Options |
StageSize.Options.sharedName(String sharedName) |
| Modifier and Type | Method and Description |
|---|---|
static StageSize |
StageSize.create(Scope scope,
List<Class<?>> dtypes,
StageSize.Options... options)
Factory method to create a class to wrap a new StageSize operation to the graph.
|
Copyright © 2015–2019. All rights reserved.