Searched refs:original_item (Results 1 – 2 of 2) sorted by relevance
418 Status AddCompressInstructions(Item* original_item, Item* compressed_item,426 Status AddRematerializedInstruction(Item* original_item, Item* remat_item);887 Status MemoryUsageTracker::AddCompressInstructions(Item* original_item, in AddCompressInstructions() argument891 memory_usage_ -= size_function_(original_item->instruction->shape()); in AddCompressInstructions()897 CHECK_EQ(original_item->buffers_output.size(), 1); in AddCompressInstructions()898 BufferId original_buffer_id = original_item->buffers_output[0]; in AddCompressInstructions()915 compressed_item->buffers_used = original_item->buffers_output; in AddCompressInstructions()937 Status MemoryUsageTracker::AddRematerializedInstruction(Item* original_item, in AddRematerializedInstruction() argument940 << original_item->instruction->name() in AddRematerializedInstruction()944 TF_RET_CHECK(original_item->placed) << original_item->instruction->name(); in AddRematerializedInstruction()[all …]
5827 def _assert_same_graph(original_item, item): argument5837 if original_item.graph is not item.graph:5839 (item, original_item))