Searched refs:id_to_instruction (Results 1 – 1 of 1) sorted by relevance
49 absl::flat_hash_map<int64, HloInstruction*> id_to_instruction; in CreateFromProto() local51 id_to_instruction[instruction->unique_id()] = instruction; in CreateFromProto()57 auto instr_it = id_to_instruction.find(instruction_id); in CreateFromProto()58 TF_RET_CHECK(instr_it != id_to_instruction.end()) in CreateFromProto()116 absl::flat_hash_map<int, HloInstruction*> id_to_instruction; in UpdateComputationSchedule() local118 InsertOrDie(&id_to_instruction, instruction->unique_id(), instruction); in UpdateComputationSchedule()186 auto it = id_to_instruction.find(id); in UpdateComputationSchedule()187 if (it == id_to_instruction.end()) { in UpdateComputationSchedule()