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 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.