Trait web_rwkv::tensor::ops::TensorCommand

pub trait TensorCommand<T: Scalar, K: Kind> {
    // Required methods
    fn copy_tensor(
        &mut self,
        source: &TensorGpu<T, ReadWrite>,
        destination: &TensorGpu<T, K>
    ) -> Result<(), TensorError>;
    fn copy_tensor_batch(
        &mut self,
        source: &TensorGpu<T, ReadWrite>,
        destination: &TensorGpu<T, K>,
        batch: usize
    ) -> Result<(), TensorError>;
}

Required Methods

fn copy_tensor( &mut self, source: &TensorGpu<T, ReadWrite>, destination: &TensorGpu<T, K> ) -> Result<(), TensorError>

fn copy_tensor_batch( &mut self, source: &TensorGpu<T, ReadWrite>, destination: &TensorGpu<T, K>, batch: usize ) -> Result<(), TensorError>

Implementations on Foreign Types

impl<T: Scalar, K: Kind> TensorCommand<T, K> for CommandEncoder

fn copy_tensor( &mut self, source: &TensorGpu<T, ReadWrite>, destination: &TensorGpu<T, K> ) -> Result<(), TensorError>

fn copy_tensor_batch( &mut self, source: &TensorGpu<T, ReadWrite>, destination: &TensorGpu<T, K>, batch: usize ) -> Result<(), TensorError>

Implementors