Enum border_tch_agent::Device
source · [−]pub enum Device {
Cpu,
Cuda(usize),
}Expand description
Device for using tch-rs.
This enum is added because tch::Device does not support serialization.
Variants
Cpu
The main CPU device.
Cuda(usize)
The main GPU device.
Trait Implementations
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error> where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error> where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more
Auto Trait Implementations
impl RefUnwindSafe for Device
impl UnwindSafe for Device
Blanket Implementations
Mutably borrows from an owned value. Read more