Searched refs:SimpleCostModel (Results 1 – 1 of 1) sorted by relevance
30 class SimpleCostModel : public ParallelCostModel { class32 SimpleCostModel(const int64 max_parallelism, in SimpleCostModel() function in xla::cpu::SimpleCostModel35 ~SimpleCostModel() override {} in ~SimpleCostModel()125 cost_model_.reset(new SimpleCostModel(max_parallelism, shape_size)); in ParallelTaskAssignment()