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()
67 const HloInstruction& hlo, const StreamAssignment& stream_assignment, in ComputeStreamToAssign()128 auto stream_assignment = absl::make_unique<StreamAssignment>(); in AssignStreams() local
52 std::unique_ptr<StreamAssignment> stream_assignment, in ThunkSchedule()
521 std::unique_ptr<StreamAssignment> stream_assignment = in RunHloPassesAndBufferAssignement() local564 std::unique_ptr<StreamAssignment> stream_assignment = in CompileModuleToLlvmIrImpl() local