Searched refs:MemoryUsageTracker (Results 1 – 1 of 1) sorted by relevance
325 class MemoryUsageTracker { class327 MemoryUsageTracker(508 MemoryUsageTracker::MemoryUsageTracker( in MemoryUsageTracker() function in xla::__anon6c3ee80d0111::MemoryUsageTracker570 Status MemoryUsageTracker::BeginInstruction(Item* item) { in BeginInstruction()597 Status MemoryUsageTracker::EndInstruction() { in EndInstruction()636 int64 MemoryUsageTracker::MemoryReducedIfRematerialized(Item* item) const { in MemoryReducedIfRematerialized()681 Status MemoryUsageTracker::AddRematerializedInstruction(Item* original_item, in AddRematerializedInstruction()749 string MemoryUsageTracker::ToString() const { in ToString()776 bool MemoryUsageTracker::Check() const { in Check()840 const MemoryUsageTracker& memory_tracker, in RematerializationCost()[all …]