Searched defs:stream_num (Results 1 – 1 of 1) sorted by relevance
37 int stream_num) { in AssignStreamToHlo()57 inline bool IsStreamNumValid(int stream_num) { in IsStreamNumValid()85 int stream_num = -1; in ComputeStreamToAssign() local104 int stream_num = stream_assignment.StreamNumberForHlo(*seen_gemm); in ComputeStreamToAssign() local111 for (int stream_num = 0; stream_num < stream_assignment.StreamCount(); in ComputeStreamToAssign() local138 int stream_num = (hlo->opcode() == HloOpcode::kRng && in AssignStreams() local