| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | ModelDataset.pbtxt | 35 name: "cpu_budget"
72 name: "cpu_budget"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ModelDataset.pbtxt | 42 name: "cpu_budget"
79 name: "cpu_budget"
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | model_dataset_op.cc | 49 model::AutotuneAlgorithm algorithm, int64 cpu_budget, in Dataset() argument
54 cpu_budget_(cpu_budget), in Dataset()
61 strings::Printf("%lld", static_cast<long long>(cpu_budget))}, in Dataset()
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | optimization_options.py | 248 cpu_budget = 0 # Indicates that all CPU cores should be used by default.
255 cpu_budget = self.autotune_cpu_budget
259 return autotune, algorithm, cpu_budget, ram_budget
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | model.h | 678 Status OptimizeLoop(AutotuneAlgorithm algorithm, int64 cpu_budget,
683 void Optimize(AutotuneAlgorithm algorithm, int64 cpu_budget, int64 ram_budget,
751 int64 cpu_budget, int64 ram_budget,
|
| D | model.cc | 1639 void Model::Optimize(AutotuneAlgorithm algorithm, int64 cpu_budget, in Optimize() argument
1648 optimization_params.set_cpu_budget(cpu_budget); in Optimize()
1694 int64 cpu_budget, int64 ram_budget, in ShouldStop() argument
1709 *cpu_budget_reached = (model_parallelism > cpu_budget); in ShouldStop()
1727 Status Model::OptimizeLoop(AutotuneAlgorithm algorithm, int64 cpu_budget, in OptimizeLoop() argument
1758 Optimize(algorithm, cpu_budget, ram_budget, /*model_input_time=*/0); in OptimizeLoop()
1813 !ShouldStop(optimization_params.cpu_budget(), in OptimizeGradientDescent()
1868 if (output_time < processing_time / optimization_params.cpu_budget() || in OptimizeHillClimb()
|
| D | model.proto | 119 int64 cpu_budget = 2; field
|
| D | model_test.cc | 961 EXPECT_EQ(optimization_params.cpu_budget(), in TEST()
962 restored_optimization_params.cpu_budget()); in TEST()
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | optimize_dataset_test.py | 612 cpu_budget = autotune_settings[2]
616 self.assertEqual(cpu_budget, 1000)
619 self.assertEqual(cpu_budget, 0)
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 378 …autotune, algorithm, cpu_budget, ram_budget = options._autotune_settings() # pylint: disable=prot…
380 dataset = _ModelDataset(dataset, algorithm, cpu_budget, ram_budget)
4647 def __init__(self, input_dataset, algorithm, cpu_budget, ram_budget): argument
4652 cpu_budget=cpu_budget,
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2545 …input_dataset\', \'output_types\', \'output_shapes\', \'algorithm\', \'cpu_budget\', \'ram_budget\…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2545 …input_dataset\', \'output_types\', \'output_shapes\', \'algorithm\', \'cpu_budget\', \'ram_budget\…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 25464 name: "cpu_budget"
|