Class ModelDataset.Options

java.lang.Object

org.tensorflow.op.data.ModelDataset.Options

Enclosing class:

ModelDataset

public static class ModelDataset.Options
extends Object

Optional attributes for ModelDataset

Method Summary

Modifier and Type	Method	Description
ModelDataset.Options	algorithm(Long algorithm)	Sets the algorithm option.
ModelDataset.Options	cpuBudget(Long cpuBudget)	Sets the cpuBudget option.
ModelDataset.Options	ramBudget(Long ramBudget)	Sets the ramBudget option.

Methods inherited from class Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Details

algorithm

public ModelDataset.Options algorithm(Long algorithm)

Sets the algorithm option.

Parameters:

algorithm - the algorithm option

Returns:

this Options instance.

cpuBudget

public ModelDataset.Options cpuBudget(Long cpuBudget)

Sets the cpuBudget option.

Parameters:

cpuBudget - the cpuBudget option

Returns:

this Options instance.

ramBudget

public ModelDataset.Options ramBudget(Long ramBudget)

Sets the ramBudget option.

Parameters:

ramBudget - the ramBudget option

Returns:

this Options instance.