pub struct UniqueConsecutiveArgs<'a, T: Element> {
pub input: TensorRef<'a, T, 2>,
pub values: TensorMut<'a, T, 2>,
pub counts: TensorMut<'a, i32, 2>,
pub counter: TensorMut<'a, i32, 1>,
}Expand description
Args bundle for a unique_consecutive launch.
Fields§
§input: TensorRef<'a, T, 2>Input [batch, row_len].
values: TensorMut<'a, T, 2>Output values [batch, max_unique] (filled left-to-right per
row up to the actual unique count).
counts: TensorMut<'a, i32, 2>Optional output per-cell counts [batch, max_unique].
counter: TensorMut<'a, i32, 1>Per-row counter [batch] — post-launch holds the actual
unique count per row.
Auto Trait Implementations§
impl<'a, T> !UnwindSafe for UniqueConsecutiveArgs<'a, T>
impl<'a, T> Freeze for UniqueConsecutiveArgs<'a, T>
impl<'a, T> RefUnwindSafe for UniqueConsecutiveArgs<'a, T>where
T: RefUnwindSafe,
impl<'a, T> Send for UniqueConsecutiveArgs<'a, T>
impl<'a, T> Sync for UniqueConsecutiveArgs<'a, T>where
T: Sync,
impl<'a, T> Unpin for UniqueConsecutiveArgs<'a, T>
impl<'a, T> UnsafeUnpin for UniqueConsecutiveArgs<'a, T>
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