Method GetStats

tensorflow/core/common_runtime/gpu/gpu_vmem_allocator.cc:109–114

}

absl::optional<AllocatorStats> GPUVMemAllocator::GetStats() {
    absl::optional<AllocatorStats> stats = device_allocator_->GetStats();
    absl::optional<AllocatorStats> allocator_stats = host_allocator_->GetStats();
    stats->peak_bytes_in_use += (allocator_stats ? allocator_stats->peak_bytes_in_use : 0);
    return stats;
}

void GPUVMemAllocator::ClearStats() {
    device_allocator_->ClearStats();