Searched defs:stream_assignment (Results 1 – 4 of 4) sorted by relevance
58 const HloModule* module, const StreamAssignment& stream_assignment, in GpuHloOrdering()193 const HloModule& module, const StreamAssignment& stream_assignment, in Build()
66 const HloInstruction& hlo, const StreamAssignment& stream_assignment, in ComputeStreamToAssign()123 auto stream_assignment = absl::make_unique<StreamAssignment>(); in AssignStreams() local
52 std::unique_ptr<StreamAssignment> stream_assignment, in ThunkSchedule()
659 std::unique_ptr<StreamAssignment> stream_assignment = AssignStreams(*module); in RunBackend() local