| Package | Description |
|---|---|
| org.tensorflow.op.data |
| Modifier and Type | Method and Description |
|---|---|
static ModelDataset.Options |
ModelDataset.algorithm(Long algorithm) |
ModelDataset.Options |
ModelDataset.Options.algorithm(Long algorithm) |
static ModelDataset.Options |
ModelDataset.cpuBudget(Long cpuBudget) |
ModelDataset.Options |
ModelDataset.Options.cpuBudget(Long cpuBudget) |
| Modifier and Type | Method and Description |
|---|---|
static ModelDataset |
ModelDataset.create(Scope scope,
Operand<?> inputDataset,
List<Class<?>> outputTypes,
List<Shape> outputShapes,
ModelDataset.Options... options)
Factory method to create a class wrapping a new ModelDataset operation.
|
Copyright © 2015–2019. All rights reserved.