Searched refs:original_item (Results 1 – 2 of 2) sorted by relevance
355 Status AddRematerializedInstruction(Item* original_item, Item* remat_item);681 Status MemoryUsageTracker::AddRematerializedInstruction(Item* original_item, in AddRematerializedInstruction() argument684 << original_item->instruction->name() in AddRematerializedInstruction()688 TF_RET_CHECK(original_item->placed) << original_item->instruction->name(); in AddRematerializedInstruction()692 remat_item->buffers_used = original_item->buffers_used; in AddRematerializedInstruction()697 for (BufferId buffer_id : original_item->buffers_used) { in AddRematerializedInstruction()711 for (BufferId old_buffer_id : original_item->buffers_defined) { in AddRematerializedInstruction()
5891 def _assert_same_graph(original_item, item): argument5901 if original_item.graph is not item.graph:5903 original_item))