Searched defs:stream_assignment (Results 1 – 5 of 5) 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()
360 std::unique_ptr<StreamAssignment> stream_assignment = AssignStreams(*module); in RunBackend() local
462 std::unique_ptr<StreamAssignment> stream_assignment = in RunBackend() local