Searched defs:ram_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()
68 int64_t ram_budget; in MakeDatasetHelper() local
|
| D | model_dataset_op.cc | 53 int64_t ram_budget) in Dataset()
59 int64_t ram_budget) in Dataset()
225 int64_t ram_budget, in MakeDatasetFromOptions()
275 bool cpu_budget, bool ram_budget, in MakeDatasetFromOptions()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | model.proto | 128 int64 ram_budget = 3; field
|
| D | dataset_options.proto | 46 int64 ram_budget = 3; field
|
| D | model.cc | 177 void RecordAutotuneRamUsage(int64 ram_budget, double max_buffered_bytes) { in RecordAutotuneRamUsage()
2169 int64_t ram_budget, double model_input_time, in Optimize()
2252 bool Model::ShouldStop(int64_t cpu_budget, int64_t ram_budget, in ShouldStop()
2278 int64_t ram_budget, in OptimizeLoop()
2582 int64_t ram_budget) { in OptimizeBuffers()
2600 bool Model::UpsizeBuffers(std::shared_ptr<Node> snapshot, int64_t ram_budget) { in UpsizeBuffers()
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | options.py | 216 ram_budget = options_lib.create_option( variable in AutotuneOptions
|