Trait DeviceTensorExt 

pub trait DeviceTensorExt {
    // Required methods
    fn to_device_tensor(&self) -> TractResult<&DeviceTensor>;
    fn as_device_tensor(&self) -> Option<&DeviceTensor>;
    fn to_device_tensor_mut(&mut self) -> TractResult<&mut DeviceTensor>;
    fn as_device_tensor_mut(&mut self) -> Option<&mut DeviceTensor>;
}

Required Methods

fn to_device_tensor(&self) -> TractResult<&DeviceTensor>

fn as_device_tensor(&self) -> Option<&DeviceTensor>

fn to_device_tensor_mut(&mut self) -> TractResult<&mut DeviceTensor>

fn as_device_tensor_mut(&mut self) -> Option<&mut DeviceTensor>

Implementations on Foreign Types

impl DeviceTensorExt for Tensor

fn to_device_tensor_mut(&mut self) -> TractResult<&mut DeviceTensor>

fn as_device_tensor_mut(&mut self) -> Option<&mut DeviceTensor>

fn to_device_tensor(&self) -> TractResult<&DeviceTensor>

fn as_device_tensor(&self) -> Option<&DeviceTensor>

Implementors