Searched refs:seen_gemms (Results 1 – 1 of 1) sorted by relevance
68 const std::vector<const HloInstruction*>& seen_gemms) { in ComputeStreamToAssign() argument103 for (const auto* seen_gemm : seen_gemms) { in ComputeStreamToAssign()127 std::vector<const HloInstruction*> seen_gemms; in AssignStreams() local142 *reachability, seen_gemms); in AssignStreams()151 seen_gemms.push_back(hlo); in AssignStreams()