pub struct TensorMapArg<R: Runtime, K: TensorMapKind> {
pub tensor: TensorArg<R>,
pub metadata: TensorMapMeta,
pub _kind: PhantomData<K>,
}Expand description
Grid constant tensor map, currently only maps to CUDA tensormap. May be interleaved or swizzled, but last dimension must be contiguous (since strides don’t include the last dimension).
The tensormap is treated as an opaque type at runtime.
Fields§
§tensor: TensorArg<R>§metadata: TensorMapMeta§_kind: PhantomData<K>Implementations§
Source§impl<R: Runtime, K: TensorMapKind> TensorMapArg<R, K>
impl<R: Runtime, K: TensorMapKind> TensorMapArg<R, K>
pub fn new(args: K::Args, tensor: TensorArg<R>, ty: impl Into<Type>) -> Self
pub fn with_elem_stride(self, elem_stride: Strides) -> Self
pub fn with_interleave(self, interleave: TensorMapInterleave) -> Self
pub fn with_swizzle(self, swizzle: TensorMapSwizzle) -> Self
pub fn with_prefetch(self, prefetch: TensorMapPrefetch) -> Self
pub fn with_nan_fill(self) -> Self
Auto Trait Implementations§
impl<R, K> !RefUnwindSafe for TensorMapArg<R, K>
impl<R, K> !UnwindSafe for TensorMapArg<R, K>
impl<R, K> Freeze for TensorMapArg<R, K>
impl<R, K> Send for TensorMapArg<R, K>
impl<R, K> Sync for TensorMapArg<R, K>
impl<R, K> Unpin for TensorMapArg<R, K>
impl<R, K> UnsafeUnpin for TensorMapArg<R, K>
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
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more