Searched refs:GetParallelTaskCount (Results 1 – 2 of 2) sorted by relevance
39 int64_t GetParallelTaskCount(HloInstruction* instruction) override { in GetParallelTaskCount() function in xla::cpu::SimpleCostModel64 int64_t GetParallelTaskCount(HloInstruction* instruction) override { in GetParallelTaskCount() function in xla::cpu::DefaultCostModel169 return cost_model_->GetParallelTaskCount(instruction); in GetTargetParallelTaskCount()
32 virtual int64_t GetParallelTaskCount(HloInstruction* instruction) = 0;