Searched refs:thunk_to_hlo_ (Results 1 – 2 of 2) sorted by relevance
37 stream_assignment_->StreamNumberForHlo(*thunk_to_hlo_.at(&thunk))) { in AddDependenciesOnTransitiveOperands()56 thunk_to_hlo_(std::move(thunk_to_hlo)) { in ThunkSchedule()63 InsertOrDie(&hlo_to_thunk, thunk_to_hlo_.at(thunk.get()), thunk.get()); in ThunkSchedule()67 const auto* dst = thunk_to_hlo_.at(thunk); in ThunkSchedule()125 stream_assignment_->StreamNumberForHlo(*thunk_to_hlo_.at(dst)); in RemoveRedundantDependencyEdges()131 stream_assignment_->StreamNumberForHlo(*thunk_to_hlo_.at(src)); in RemoveRedundantDependencyEdges()174 if (thunk_to_hlo_.at(thunk) != nullptr) { in ToString()175 absl::StrAppend(&result, thunk_to_hlo_.at(thunk)->ToString()); in ToString()185 absl::StrAppend(&result, "\t", thunk_to_hlo_.at(dependent)->name(), in ToString()186 " depends on ", thunk_to_hlo_.at(dependency)->name(), in ToString()
80 return stream_assignment_->StreamNumberForHlo(*thunk_to_hlo_.at(thunk)); in StreamNumberForThunk()109 absl::flat_hash_map<const Thunk*, const HloInstruction*> thunk_to_hlo_; variable