Struct tvm::function::ffi::DLDevice

#[repr(C)]
pub struct DLDevice {
    pub device_type: u32,
    pub device_id: i32,
}

\brief A Device for Tensor and operator.

Fields

device_type: u32

\brief The device type used in the device.

device_id: i32

\brief The device index

Implementations

impl DLDevice

pub fn cpu(device_id: usize) -> DLDevice

pub fn llvm(device_id: usize) -> DLDevice

pub fn stackvm(device_id: usize) -> DLDevice

pub fn cuda(device_id: usize) -> DLDevice

pub fn nvptx(device_id: usize) -> DLDevice

pub fn cl(device_id: usize) -> DLDevice

pub fn metal(device_id: usize) -> DLDevice

pub fn vpi(device_id: usize) -> DLDevice

pub fn rocm(device_id: usize) -> DLDevice

pub fn ext_dev(device_id: usize) -> DLDevice

Trait Implementations

impl Clone for DLDevice

pub fn clone(&self) -> DLDevice

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for DLDevice

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

Formats the value using the given formatter.

impl Default for DLDevice

pub fn default() -> DLDevice

Returns the “default value” for a type.

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

pub fn from(val: &'a DLDevice) -> ArgValue<'a>

Performs the conversion.

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

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

Performs the conversion.

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

pub fn from(val: DLDevice) -> ArgValue<'a>

Performs the conversion.

impl From<DLDevice> for TVMValue

pub fn from(val: DLDevice) -> TVMValue

Performs the conversion.

impl From<DLDevice> for RetValue

pub fn from(val: DLDevice) -> RetValue

Performs the conversion.

impl From<DLDevice> for Device

pub fn from(dev: DLDevice) -> Device

Performs the conversion.

impl From<Device> for DLDevice

pub fn from(dev: Device) -> DLDevice

Performs the conversion.

impl From<TVMValue> for DLDevice

pub fn from(val: TVMValue) -> DLDevice

Performs the conversion.

impl FromStr for DLDevice

Creates a DLDevice 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<DLDevice, <DLDevice as FromStr>::Err>

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

impl PartialEq<DLDevice> for DLDevice

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

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

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

This method tests for !=.

impl<'a, 'v> TryFrom<&'a ArgValue<'v>> for DLDevice

type Error = ValueDowncastError

The type returned in the event of a conversion error.

pub fn try_from(
    val: &'a ArgValue<'v>
) -> Result<DLDevice, <DLDevice as TryFrom<&'a ArgValue<'v>>>::Error>

Performs the conversion.

impl<'a> TryFrom<ArgValue<'a>> for DLDevice

type Error = ValueDowncastError

The type returned in the event of a conversion error.

pub fn try_from(
    val: ArgValue<'a>
) -> Result<DLDevice, <DLDevice as TryFrom<ArgValue<'a>>>::Error>

Performs the conversion.

impl TryFrom<RetValue> for DLDevice

type Error = ValueDowncastError

The type returned in the event of a conversion error.

pub fn try_from(
    val: RetValue
) -> Result<DLDevice, <DLDevice as TryFrom<RetValue>>::Error>

Performs the conversion.

impl Copy for DLDevice

impl Eq for DLDevice

impl StructuralEq for DLDevice

impl StructuralPartialEq for DLDevice