Searched refs:indirect_user (Results 1 – 1 of 1) sorted by relevance
1150 for (Item* indirect_user : indirect_users) { in AddRematerializedInstruction()1153 instruction_list_.GetItem(indirect_user->instruction->operand(0)); in AddRematerializedInstruction()1154 switch (indirect_user->instruction->opcode()) { in AddRematerializedInstruction()1160 indirect_user->buffers_used = source_item->buffers_output; in AddRematerializedInstruction()1161 indirect_user->buffers_output = source_item->buffers_output; in AddRematerializedInstruction()1165 indirect_user->buffers_used = source_item->buffers_defined; in AddRematerializedInstruction()1166 indirect_user->buffers_output = source_item->buffers_defined; in AddRematerializedInstruction()1174 Cast<HloGetTupleElementInstruction>(indirect_user->instruction); in AddRematerializedInstruction()1178 indirect_user->buffers_output.push_back(buffer_id); in AddRematerializedInstruction()1182 indirect_user->buffers_used.push_back(buffer_id); in AddRematerializedInstruction()[all …]