Searched refs:instruction_schedule_ (Results 1 – 4 of 4) sorted by relevance
116 if (instruction_schedule_.count(instruction) != 0) { in FlattenSchedule()119 instruction_schedule_.insert({instruction, time++}); in FlattenSchedule()124 DCHECK_EQ(instruction_schedule_.size(), in FlattenSchedule()126 DCHECK_LE(instruction_schedule_.size(), time); in FlattenSchedule()133 if (instruction_schedule_.count(value->defining_instruction()) == 0) { in CalculateBufferStartEndMap()137 int64 buffer_start_time = instruction_schedule_[value->instruction()]; in CalculateBufferStartEndMap()152 if (instruction_schedule_.count(used) == 0) { in CalculateBufferStartEndMap()158 buffer_end_time = std::max(buffer_end_time, instruction_schedule_[used]); in CalculateBufferStartEndMap()
73 return instruction_schedule_; in instruction_schedule()204 absl::flat_hash_map<const HloInstruction*, int64> instruction_schedule_; variable
248 instruction_schedule_ = nullptr;
387 instruction_schedule_ = in CostAnalysisPrefetchIntervalPicker()394 std::vector<float> instructions_elapsed_time(instruction_schedule_->size(), in CostAnalysisPrefetchIntervalPicker()396 for (const auto& instruction_and_logical_time : *instruction_schedule_) { in CostAnalysisPrefetchIntervalPicker()