Searched refs:MemoryReducedIfRematerialized (Results 1 – 1 of 1) sorted by relevance
348 int64 MemoryReducedIfRematerialized(Item* item) const;636 int64 MemoryUsageTracker::MemoryReducedIfRematerialized(Item* item) const { in MemoryReducedIfRematerialized() function in xla::__anon6c3ee80d0111::MemoryUsageTracker909 memory_tracker.MemoryReducedIfRematerialized(item); in PickRematerializationCandidate()1046 memory_tracker.MemoryReducedIfRematerialized(best_item)) in RematerializeComputation()