Searched refs:GetParallelTaskCount (Results 1 – 2 of 2) sorted by relevance
37 int64 GetParallelTaskCount(HloInstruction* instruction) override { in GetParallelTaskCount() function in xla::cpu::SimpleCostModel61 int64 GetParallelTaskCount(HloInstruction* instruction) override { in GetParallelTaskCount() function in xla::cpu::DefaultCostModel156 return cost_model_->GetParallelTaskCount(instruction); in GetTargetParallelTaskCount()
31 virtual int64 GetParallelTaskCount(HloInstruction* instruction) = 0;