Struct RawTensor

pub struct RawTensor<'r> { /* private fields */ }

A raw tensor type that doesn’t have any guarantees and will be unconditionally dropped

Implementations

impl RawTensor<'_>

pub fn create_host_tensor_from_device( &self, copy_data: bool, ) -> RawTensor<'static>

Creates a new host tensor from the device tensor

pub fn copy_from_host_tensor(&mut self, tensor: &RawTensor<'_>) -> Result<()>

Copies the data from a host tensor to the self tensor

pub fn copy_to_host_tensor(&self, tensor: &mut RawTensor<'_>) -> Result<()>

Copies the data from the self tensor to a host tensor

pub fn shape(&self) -> TensorShape

Returns the shape of the tensor

pub fn get_dimension_type(&self) -> DimensionType

Returns the dimension type of the tensor

pub fn destroy(self)

Cleans up the tensor by calling the destructor of the tensor

pub fn size(&self) -> usize

Returns the size of the tensor when counted by bytes

pub fn element_size(&self) -> usize

Returns the size of the tensor when counted by elements

pub fn dimensions(&self) -> usize

Returns the number of dimensions of the tensor

pub fn width(&self) -> u32

Returns the width of the tensor

pub fn height(&self) -> u32

Returns the height of the tensor

pub fn channel(&self) -> u32

Returns the channel of the tensor

pub fn is_dynamic_unsized(&self) -> bool

Returns true if the tensor is unsized and dynamic (needs to be resized to work)

pub fn wait(&self, map_type: MapType, finish: bool)

Waits for the tensor to be ready

pub unsafe fn unchecked_host_ptr(&self) -> *mut c_void

Safety

This is very unsafe do not use this unless you know what you are doing Gives a raw pointer to the tensor’s data P.S. I don’t know what I’m doing

pub unsafe fn unchecked_host_bytes(&mut self) -> &mut [u8]

Safety

This is very unsafe do not use this unless you know what you are doing Gives a mutable byte slice to the tensor’s data

pub unsafe fn to_concrete<T: TensorType>(self) -> Tensor<T>

where T::H: HalideType,

Safety

This is very unsafe do not use this unless you know what you are doing