Struct GpuTensorViewMut

pub struct GpuTensorViewMut<'a> {
    pub shape_strides: ShapeStrides,
    /* private fields */
}

Same as GpuTensorView but mutable

Fields

shape_strides: ShapeStrides

Implementations

impl<'a> GpuTensorViewMut<'a>

pub fn from_tensor( gpu_tensor: &'a mut GpuTensor, dim_strides: ShapeStrides, ) -> Self

impl<'a> GpuTensorViewMut<'a>

pub async fn assign_kernel(&mut self, data: f32)

Trait Implementations

impl<'a> GpuAllocated for GpuTensorViewMut<'a>

fn get_gpu(&self) -> &'static GpuInstance

fn internal_gpu_buffer(&self) -> &GpuBuffer

fn internal_buffer_size_in_bytes(&self) -> usize

impl<'a> MutShapeStrideTrait for GpuTensorViewMut<'a>

fn increase_rank(&mut self)

Artificially increase the rank, setting the new dimension shape to 1

fn decrease_rank(&mut self)

Decrease the rank if the leading shape dimension is 1, otherwise panics

impl<'a> ShapeStrideTrait for GpuTensorViewMut<'a>

fn shape(&self) -> &VecDeque<usize>

fn strides(&self) -> &VecDeque<usize>

fn offset(&self) -> usize

fn rank(&self) -> usize

fn numel(&self) -> usize

fn numel_from_shape(shape: &VecDeque<usize>) -> usize