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