Searched refs:stream_assignment_ (Results 1 – 2 of 2) sorted by relevance
73 if (stream_assignment_) { in StreamCount()74 return stream_assignment_->StreamCount(); in StreamCount()79 if (stream_assignment_) { in StreamNumberForThunk()80 return stream_assignment_->StreamNumberForHlo(*thunk_to_hlo_.at(thunk)); in StreamNumberForThunk()107 std::unique_ptr<StreamAssignment> stream_assignment_; variable
36 if (stream_assignment_->StreamNumberForHlo(operand) != in AddDependenciesOnTransitiveOperands()37 stream_assignment_->StreamNumberForHlo(*thunk_to_hlo_.at(&thunk))) { in AddDependenciesOnTransitiveOperands()55 stream_assignment_(std::move(stream_assignment)), in ThunkSchedule()68 CHECK(stream_assignment_->HasStreamAssigned(*dst)); in ThunkSchedule()97 int stream_count = stream_assignment_->StreamCount(); in RemoveRedundantDependencyEdges()125 stream_assignment_->StreamNumberForHlo(*thunk_to_hlo_.at(dst)); in RemoveRedundantDependencyEdges()131 stream_assignment_->StreamNumberForHlo(*thunk_to_hlo_.at(src)); in RemoveRedundantDependencyEdges()