Searched refs:MemoryUsageTracker (Results 1 – 1 of 1) sorted by relevance
491 class MemoryUsageTracker { class493 MemoryUsageTracker(781 MemoryUsageTracker::MemoryUsageTracker( in MemoryUsageTracker() function in xla::__anon95ba67740111::MemoryUsageTracker861 Status MemoryUsageTracker::BeginInstruction(Item* item) { in BeginInstruction()888 Status MemoryUsageTracker::EndInstruction() { in EndInstruction()929 int64 MemoryUsageTracker::MemoryReducedIfCompressed( in MemoryReducedIfCompressed()953 int64 MemoryUsageTracker::MemoryReducedIfRematerialized( in MemoryReducedIfRematerialized()1011 Status MemoryUsageTracker::AddCompressInstructions(Item* original_item, in AddCompressInstructions()1066 Status MemoryUsageTracker::AddRematerializedInstruction( in AddRematerializedInstruction()1210 string MemoryUsageTracker::ToString() const { in ToString()[all …]