Trait DeviceTensorExt 

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

Required Methods

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

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

Implementations on Foreign Types

impl DeviceTensorExt for Tensor

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

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

Implementors