Searched refs:ram_budget (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | model_dataset_op.cc | 50 int64 ram_budget) in Dataset() argument
55 ram_budget_(ram_budget), in Dataset()
63 strings::Printf("%lldB", static_cast<long long>(ram_budget))}}) { in Dataset()
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | optimization_options.py | 249 ram_budget = 0 # Indicates that default value of RAM budget should be used.
257 ram_budget = self.autotune_ram_budget
259 return autotune, algorithm, cpu_budget, ram_budget
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | ModelDataset.pbtxt | 86 name: "ram_budget"
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | model.h | 679 int64 ram_budget, CancellationManager* cancellation_mgr);
683 void Optimize(AutotuneAlgorithm algorithm, int64 cpu_budget, int64 ram_budget,
751 int64 cpu_budget, int64 ram_budget,
|
| D | model.cc | 1640 int64 ram_budget, double model_input_time) { in Optimize() argument
1649 optimization_params.set_ram_budget(ram_budget); in Optimize()
1694 int64 cpu_budget, int64 ram_budget, in ShouldStop() argument
1723 return all_max || TotalMaximumBufferedBytes(snapshot) > ram_budget; in ShouldStop()
1728 int64 ram_budget, in OptimizeLoop() argument
1758 Optimize(algorithm, cpu_budget, ram_budget, /*model_input_time=*/0); in OptimizeLoop()
1814 optimization_params.ram_budget(), parameters, in OptimizeGradientDescent()
1871 optimization_params.ram_budget()) { in OptimizeHillClimb()
|
| D | model.proto | 122 int64 ram_budget = 3; field
|
| D | model_test.cc | 963 EXPECT_EQ(optimization_params.ram_budget(), in TEST()
964 restored_optimization_params.ram_budget()); in TEST()
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | optimize_dataset_test.py | 613 ram_budget = autotune_settings[3]
617 self.assertEqual(ram_budget, 999999999)
620 self.assertEqual(ram_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
4653 ram_budget=ram_budget,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2545 … \'output_types\', \'output_shapes\', \'algorithm\', \'cpu_budget\', \'ram_budget\', \'name\'], va…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2545 … \'output_types\', \'output_shapes\', \'algorithm\', \'cpu_budget\', \'ram_budget\', \'name\'], va…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 25471 name: "ram_budget"
|