Searched refs:cost_model_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | scheduler.cc | 75 : graph_(g), cost_model_(cost_model) {} in SlackAnalysis()
91 Microseconds ctime = cost_model_->TimeEstimate(curr); in ComputeAsap()
160 Microseconds ctime = cost_model_->TimeEstimate(src); in ComputeAlap()
193 cost_model_(cost_model), in GreedyScheduler()
269 e.time = event.time + cost_model_->TimeEstimate(e.node); in ComputeSchedule()
296 : devices_(devices), cost_model_(cost_model), graph_(g) {} in PriorityScheduler()
301 SlackAnalysis slack(graph_, cost_model_); in ComputeSchedule()
303 GreedyScheduler greedysched(devices_, cost_model_, graph_, &slacks); in ComputeSchedule()
|
| D | scheduler.h | 50 const CostModel* cost_model_; variable
85 const CostModel* cost_model_; variable
110 const CostModel* cost_model_; variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | parallel_task_assignment.cc | 119 cost_model_.reset(new DefaultCostModel(max_parallelism, shape_size, in ParallelTaskAssignment()
125 cost_model_.reset(new SimpleCostModel(max_parallelism, shape_size)); in ParallelTaskAssignment()
156 return cost_model_->GetParallelTaskCount(instruction); in GetTargetParallelTaskCount()
|
| D | parallel_task_assignment.h | 51 std::unique_ptr<ParallelCostModel> cost_model_;
|