Searched refs:schedule_after_ (Results 1 – 2 of 2) sorted by relevance
889 absl::flat_hash_map<int64, std::vector<HloInstruction*>> schedule_after_; variable
3170 schedule_after_[copy_allocation->copy_start_schedule_after()].push_back( in ScheduleAsynchronousCopies()3227 auto insts_after_iter = schedule_after_.find(instruction_index); in FixSchedule()3228 if (insts_after_iter != schedule_after_.end()) { in FixSchedule()