Struct TPUInfo

pub struct TPUInfo {

    pub vendor: TPUVendor,
    pub model_name: String,
    pub architecture: TPUArchitecture,
    pub connection_type: TPUConnectionType,
    pub tops_performance: Option<f32>,
    pub memory_gb: Option<f32>,
    pub memory_bandwidth_gbps: Option<f32>,
    pub core_count: Option<u32>,
    pub driver_version: Option<String>,
    pub firmware_version: Option<String>,
    pub device_id: Option<String>,
    pub supported_frameworks: Vec<String>,
    pub power_consumption: Option<f32>,
    pub temperature: Option<f32>,
    pub clock_frequency: Option<u32>,
    pub supported_dtypes: Vec<String>,
    pub capabilities: HashMap<String, String>,
}

TPU information structure

Fields

vendor: TPUVendor

TPU vendor

model_name: String

TPU model name

architecture: TPUArchitecture

TPU architecture/generation

connection_type: TPUConnectionType

Connection type

tops_performance: Option<f32>

Performance in TOPS (Tera Operations Per Second)

memory_gb: Option<f32>

Memory size in GB

memory_bandwidth_gbps: Option<f32>

Memory bandwidth in GB/s

core_count: Option<u32>

Number of cores/processing units

driver_version: Option<String>

Driver version

firmware_version: Option<String>

Firmware version

device_id: Option<String>

Device ID

supported_frameworks: Vec<String>

Supported frameworks

power_consumption: Option<f32>

Power consumption in watts

temperature: Option<f32>

Operating temperature in Celsius

clock_frequency: Option<u32>

Clock frequency in MHz

supported_dtypes: Vec<String>

Supported data types

capabilities: HashMap<String, String>

Additional capabilities

Implementations

impl TPUInfo

pub fn query_all() -> Result<Vec<TPUInfo>>

Query all available TPUs in the system

pub fn vendor(&self) -> &TPUVendor

Get TPU vendor

pub fn model_name(&self) -> &str

Get TPU model name

pub fn architecture(&self) -> &TPUArchitecture

Get TPU architecture

pub fn tops_performance(&self) -> Option<f32>

Get performance in TOPS

pub fn supports_framework(&self, framework: &str) -> bool

Check if TPU supports a specific framework

pub fn supports_dtype(&self, dtype: &str) -> bool

Check if TPU supports a specific data type

Trait Implementations

impl Clone for TPUInfo

fn clone(&self) -> TPUInfo

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for TPUInfo

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for TPUInfo

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for TPUInfo

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.