Searched refs:SimpleCostModel (Results 1 - 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ | ||
H A D | parallel_task_assignment.cc | 28 class SimpleCostModel : public ParallelCostModel { class in namespace:xla::cpu 30 SimpleCostModel(const int64 max_parallelism, function in class:xla::cpu::SimpleCostModel 33 ~SimpleCostModel() override {} 121 cost_model_.reset(new SimpleCostModel(max_parallelism, shape_size)); |
Completed in 166 milliseconds