Searched refs:seen_gemms (Results 1 – 1 of 1) sorted by relevance
69 const std::vector<const HloInstruction*>& seen_gemms) { in ComputeStreamToAssign() argument108 for (const auto* seen_gemm : seen_gemms) { in ComputeStreamToAssign()132 std::vector<const HloInstruction*> seen_gemms; in AssignStreams() local147 *reachability, seen_gemms); in AssignStreams()156 seen_gemms.push_back(hlo); in AssignStreams()