Searched refs:memory_usage_ (Results 1 – 1 of 1) sorted by relevance
365 int64 memory_usage() const { return memory_usage_; } in memory_usage()498 int64 memory_usage_ = 0; member in xla::__anon6c3ee80d0111::MemoryUsageTracker582 memory_usage_ += AllocatedSize(buffer_id); in BeginInstruction()588 VLOG(3) << " memory usage = " << memory_usage_; in BeginInstruction()609 memory_usage_ -= AllocatedSize(buffer_id); in EndInstruction()610 CHECK_GE(memory_usage_, 0); in EndInstruction()620 memory_usage_ -= AllocatedSize(buffer_id); in EndInstruction()621 CHECK_GE(memory_usage_, 0); in EndInstruction()627 VLOG(3) << " memory usage = " << memory_usage_; in EndInstruction()701 memory_usage_ += AllocatedSize(buffer.id); in AddRematerializedInstruction()[all …]