Searched refs:instruction_index (Results 1 – 6 of 6) sorted by relevance
252 uint32_t instruction_index = 1; in TryToAddFunction() local258 while (instruction_index < num_instructions && in TryToAddFunction()259 message_.instruction(instruction_index).opcode() == in TryToAddFunction()262 ir_context, message_.instruction(instruction_index))); in TryToAddFunction()263 instruction_index++; in TryToAddFunction()267 if (instruction_index == num_instructions || in TryToAddFunction()268 message_.instruction(instruction_index).opcode() != SpvOpLabel) { in TryToAddFunction()274 while (instruction_index < num_instructions && in TryToAddFunction()275 message_.instruction(instruction_index).opcode() != SpvOpFunctionEnd) { in TryToAddFunction()277 assert(message_.instruction(instruction_index).opcode() == SpvOpLabel); in TryToAddFunction()[all …]
3199 for (int64 instruction_index = 0; in FixSchedule() local3200 instruction_index < flattened_instructions_.size(); in FixSchedule()3201 ++instruction_index) { in FixSchedule()3202 auto insts_before_iter = schedule_before_.find(instruction_index); in FixSchedule()3206 VLOG(4) << "before " << instruction_index << ": " in FixSchedule()3213 HloInstruction* instruction = flattened_instructions_[instruction_index]; in FixSchedule()3223 VLOG(4) << "inst " << instruction_index << ": " << instruction->name(); in FixSchedule()3227 auto insts_after_iter = schedule_after_.find(instruction_index); in FixSchedule()3231 VLOG(4) << "after " << instruction_index << ": " in FixSchedule()
1874 int64 instruction_index = 0; in RematerializeComputation() local1884 << ", callee usage = " << callee_usage << ", [" << instruction_index in RematerializeComputation()1886 instruction_index++; in RematerializeComputation()
959 uint32_t instruction_index = 0; in GenerateSourceMessages() local970 if (instruction_index == debug_record[kInstCommonOutInstructionIdx]) { in GenerateSourceMessages()973 instruction_index++; in GenerateSourceMessages()