Searched refs:stream_assignment_ (Results 1 – 2 of 2) sorted by relevance
36 if (stream_assignment_->StreamNumberForHlo(operand) != in AddDependenciesOnTransitiveOperands()37 stream_assignment_->StreamNumberForHlo(*thunk.hlo_instruction())) { in AddDependenciesOnTransitiveOperands()55 stream_assignment_(std::move(stream_assignment)) { in ThunkSchedule()69 CHECK(stream_assignment_->HasStreamAssigned(*dst)); in ThunkSchedule()91 int stream_count = stream_assignment_->StreamCount(); in RemoveRedundantDependencyEdges()119 stream_assignment_->StreamNumberForHlo(*dst->hlo_instruction()); in RemoveRedundantDependencyEdges()125 stream_assignment_->StreamNumberForHlo(*src->hlo_instruction()); in RemoveRedundantDependencyEdges()
64 int StreamCount() const { return stream_assignment_->StreamCount(); } in StreamCount()66 return stream_assignment_->StreamNumberForHlo(hlo); in StreamNumberForHlo()91 std::unique_ptr<StreamAssignment> stream_assignment_; variable