Struct tvm::Device

pub struct Device {
    pub device_type: DeviceType,
    pub device_id: usize,
}

Fields

device_type: DeviceType

device_id: usize

Implementations

impl Device

pub fn new(device_type: DeviceType, device_id: usize) -> Device

impl Device

pub fn cpu(device_id: usize) -> Device

pub fn llvm(device_id: usize) -> Device

pub fn stackvm(device_id: usize) -> Device

pub fn cuda(device_id: usize) -> Device

pub fn nvptx(device_id: usize) -> Device

pub fn cl(device_id: usize) -> Device

pub fn metal(device_id: usize) -> Device

pub fn vpi(device_id: usize) -> Device

pub fn rocm(device_id: usize) -> Device

pub fn ext_dev(device_id: usize) -> Device

Trait Implementations

impl Clone for Device

pub fn clone(&self) -> Device

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Device

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

Formats the value using the given formatter.

impl Default for Device

pub fn default() -> Device

Returns the “default value” for a type.

impl Display for Device

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

Formats the value using the given formatter.

impl<'a> From<&'a Device> for DLDevice

pub fn from(dev: &'a Device) -> DLDevice

Performs the conversion.

impl<'a> From<&'a Device> for ArgValue<'a>

pub fn from(dev: &'a Device) -> ArgValue<'a>

Performs the conversion.

impl<'a> From<&'a str> for Device

pub fn from(target: &str) -> Device

Performs the conversion.

impl From<DLDevice> for Device

pub fn from(dev: DLDevice) -> Device

Performs the conversion.

impl<'a> From<Device> for ArgValue<'a>

pub fn from(dev: Device) -> ArgValue<'a>

Performs the conversion.

impl From<Device> for DLDevice

pub fn from(dev: Device) -> DLDevice

Performs the conversion.

impl From<Device> for RetValue

pub fn from(ret_value: Device) -> RetValue

Performs the conversion.

impl FromStr for Device

Creates a Device from a string (e.g., “cpu”, “cuda”, “ext_dev”)

type Err = UnsupportedDeviceError

The associated error which can be returned from parsing.

pub fn from_str(type_str: &str) -> Result<Device, <Device as FromStr>::Err>

Parses a string s to return a value of this type.

impl Hash for Device

pub fn hash<__H>(&self, state: &mut __H) where
    __H: Hasher, 

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H) where
    H: Hasher, 

Feeds a slice of this type into the given Hasher.

impl PartialEq<Device> for Device

pub fn eq(&self, other: &Device) -> bool

This method tests for self and other values to be equal, and is used by ==.

pub fn ne(&self, other: &Device) -> bool

This method tests for !=.

impl TryFrom<RetValue> for Device

type Error = Error

The type returned in the event of a conversion error.

pub fn try_from(ret_value: RetValue) -> Result<Device, Error>

Performs the conversion.

impl Copy for Device

impl Eq for Device

impl StructuralEq for Device

impl StructuralPartialEq for Device