pub enum KernelArg {
Tensor(TensorRef),
Buffer {
ptr: *const u8,
size: usize,
},
Scalar(ScalarValue),
LocalMemory(usize),
}Expand description
Kernel argument types
Variants§
Tensor(TensorRef)
Tensor reference
Buffer
Raw memory buffer
Scalar(ScalarValue)
Scalar value
LocalMemory(usize)
Local/shared memory allocation
Trait Implementations§
Auto Trait Implementations§
impl Freeze for KernelArg
impl !RefUnwindSafe for KernelArg
impl !Send for KernelArg
impl !Sync for KernelArg
impl Unpin for KernelArg
impl UnsafeUnpin for KernelArg
impl !UnwindSafe for KernelArg
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more