Searched refs:cpu_budget (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | model_dataset_op.cc | 52 Dataset(OpKernelContext* ctx, const DatasetBase* input, int64 cpu_budget) in Dataset() argument
55 cpu_budget_(cpu_budget) { in Dataset()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | model.cc | 410 void Model::Optimize(int64 cpu_budget) { in Optimize() argument
430 if (output_time < processing_time / cpu_budget || all_max) { in Optimize()
|
| D | model.h | 390 void Optimize(int64 cpu_budget) LOCKS_EXCLUDED(mu_);
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 197 cpu_budget = 0 # Indicates that all CPU cores should be used.
202 cpu_budget = options.experimental_optimization.autotune_cpu_budget
205 dataset = _ModelDataset(dataset, cpu_budget)
3286 def __init__(self, input_dataset, cpu_budget): argument
3290 cpu_budget=cpu_budget,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 2045 …argspec: "args=[\'input_dataset\', \'output_types\', \'output_shapes\', \'cpu_budget\', \'name\'],…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 2045 …argspec: "args=[\'input_dataset\', \'output_types\', \'output_shapes\', \'cpu_budget\', \'name\'],…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 19949 name: "cpu_budget"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 40730 name: "cpu_budget"
|