Searched refs:HasStreamAssigned (Results 1 – 4 of 4) sorted by relevance
29 bool StreamAssignment::HasStreamAssigned(const HloInstruction& hlo) const { in HasStreamAssigned() function in xla::gpu::StreamAssignment92 if (stream_assignment.HasStreamAssigned(*operand)) { in ComputeStreamToAssign()
31 bool HasStreamAssigned(const HloInstruction& hlo) const;
88 if (stream_assignment.HasStreamAssigned(*hlo)) { in GpuHloOrdering()
68 CHECK(stream_assignment_->HasStreamAssigned(*dst)); in ThunkSchedule()