Skip to content

tenzr::HardwareManager::DeviceMemoryStats

Device allocator statistics (from PJRT_Device_MemoryStats when the plugin supports it).

#include <hardware_manager.h>

Public Attributes

Name
bool available
int64_t bytes_in_use
int64_t peak_bytes_in_use
int64_t bytes_limit

Public Attributes Documentation

variable available

bool available {false};

variable bytes_in_use

int64_t bytes_in_use {0};

variable peak_bytes_in_use

int64_t peak_bytes_in_use {0};

variable bytes_limit

int64_t bytes_limit {0};