Searched refs:remat_item (Results 1 – 1 of 1) sorted by relevance
355 Status AddRematerializedInstruction(Item* original_item, Item* remat_item);428 Buffer& RematerializeBuffer(const Buffer& original_buffer, Item* remat_item, in RematerializeBuffer() argument437 return NewBuffer(remat_item, original_buffer.size, in RematerializeBuffer()682 Item* remat_item) { in AddRematerializedInstruction() argument685 << ", remat_instruction = " << remat_item->instruction->name(); in AddRematerializedInstruction()689 TF_RET_CHECK(!remat_item->placed) << remat_item->instruction->name(); in AddRematerializedInstruction()692 remat_item->buffers_used = original_item->buffers_used; in AddRematerializedInstruction()705 buffer.users.push_back(remat_item); in AddRematerializedInstruction()731 RematerializeBuffer(old_buffer, remat_item, std::move(unplaced_users)); in AddRematerializedInstruction()733 remat_item->buffers_defined.push_back(new_buffer.id); in AddRematerializedInstruction()[all …]