Searched defs:cpu_budget (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | finalize_dataset_op.cc | 40 int64_t* cpu_budget, int64_t* ram_budget) { in GetModelDatasetParams()
67 int64_t cpu_budget; in MakeDatasetHelper() local
|
| D | model_dataset_op.cc | 52 model::AutotuneAlgorithm algorithm, int64_t cpu_budget, in Dataset()
58 model::AutotuneAlgorithm algorithm, int64_t cpu_budget, in Dataset()
224 int64_t cpu_budget, in MakeDatasetFromOptions()
275 bool cpu_budget, bool ram_budget, in MakeDatasetFromOptions()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | model.proto | 125 int64 cpu_budget = 2; field
|
| D | dataset_options.proto | 40 int32 cpu_budget = 2; field
|
| D | model.cc | 2168 void Model::Optimize(AutotuneAlgorithm algorithm, int64_t cpu_budget, in Optimize()
2252 bool Model::ShouldStop(int64_t cpu_budget, int64_t ram_budget, in ShouldStop()
2277 Status Model::OptimizeLoop(AutotuneAlgorithm algorithm, int64_t cpu_budget, in OptimizeLoop()
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | options.py | 208 cpu_budget = options_lib.create_option( variable in AutotuneOptions
|