Searched refs:MemoryReducedIfRematerialized (Results 1 – 1 of 1) sorted by relevance
544 int64 MemoryReducedIfRematerialized(953 int64 MemoryUsageTracker::MemoryReducedIfRematerialized( in MemoryReducedIfRematerialized() function in xla::__anon95ba67740111::MemoryUsageTracker1467 const int64 memory_reduced = MemoryReducedIfRematerialized(block); in PickRematerializationCandidates()1531 memory_tracker->MemoryReducedIfRematerialized(*best_items); in RematerializeInstructions()