Searched refs:AddRematerializedInstruction (Results 1 – 1 of 1) sorted by relevance
555 Status AddRematerializedInstruction(Item* original_item, Item* remat_item,1066 Status MemoryUsageTracker::AddRematerializedInstruction( in AddRematerializedInstruction() function in xla::__anon95ba67740111::MemoryUsageTracker1596 TF_RETURN_IF_ERROR(memory_tracker->AddRematerializedInstruction( in RematerializeInstructions()