Enum IValueKind

pub enum IValueKind {
    Tensor,
    Bool,
    Int,
    IntList,
    Double,
    String,
    TensorList,
    Device,
    None,
    Other,
}

Enum representing the different internal types an IValue can hold.

Check each variant docs to see what the internal storage in C++ is, and what the cost of moving across the Rust<>C++ boundary is.

Variants

Tensor

• C++ type: at::Tensor
• Rust type: Tensor

Bool

• C++ type: bool
• Rust type: bool

Int

• C++ type: int64_t
• Rust type: i64

IntList

• C++ type: c10::List<int64_t>
• Rust type: Vec<i64>

Passing across the C++-Rust boundary will copy the vector.

Double

• C++ type: double
• Rust type: f64

String

• C++ type: c10::intrusive_ptr<ConstantString>
• Rust type: String

Passing across the C++-Rust boundary will copy the string.

TensorList

• C++ type: c10::List<at::Tensor>
• Rust type: Vec<Tensor>

Passing across the C++-Rust boundary will copy the vector.

Device

• C++ type: c10::Device
• Rust type: Device

None

Other

Catch-all for all other types. This is used for types that are not natively supported in rust and can remain as opaque IValues for interacting with torch apis. There is an overhead associated with tracking alias and borrows for any trivial IValues being converted to this type so they should be natively supported. Most of them are already supported.

Trait Implementations

impl Clone for IValueKind

fn clone(&self) -> IValueKind

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for IValueKind

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

Formats the value using the given formatter.

impl PartialEq for IValueKind

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

Tests for self and other values to be equal, and is used by ==.

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

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for IValueKind

impl Eq for IValueKind

impl StructuralPartialEq for IValueKind