Searched defs:stream_num (Results 1 – 2 of 2) sorted by relevance
38 int stream_num) { in AssignStreamToHlo()58 inline bool IsStreamNumValid(int stream_num) { in IsStreamNumValid()90 int stream_num = -1; in ComputeStreamToAssign() local109 int stream_num = stream_assignment.StreamNumberForHlo(*seen_gemm); in ComputeStreamToAssign() local116 for (int stream_num = 0; stream_num < stream_assignment.StreamCount(); in ComputeStreamToAssign() local143 int stream_num = (hlo->opcode() == HloOpcode::kRng && in AssignStreams() local
44 for (size_t stream_num = 0; stream_num < encoder_config.number_of_streams; in CreateVideoStreams() local