Skip to main content

TensorDeviceTransfer

Trait TensorDeviceTransfer 

Source
pub trait TensorDeviceTransfer {
    // Provided methods
    fn download_to_host(&mut self, tensor: &Tensor) -> Result<Tensor> { ... }
    fn upload_host_tensor(&mut self, tensor: &Tensor) -> Result<Tensor> { ... }
}
Expand description

Device transfer operations on backend boundaries.

§Examples

use tenferro_tensor::TensorDeviceTransfer;

fn accepts_transfer<B: TensorDeviceTransfer>(_backend: &mut B) {}

Provided Methods§

Source

fn download_to_host(&mut self, tensor: &Tensor) -> Result<Tensor>

Source

fn upload_host_tensor(&mut self, tensor: &Tensor) -> Result<Tensor>

Implementors§