Struct UniqueArgs 

pub struct UniqueArgs<'a, T: Element> {
    pub input: TensorRef<'a, T, 2>,
    pub sorted_scratch: TensorMut<'a, T, 2>,
    pub sorted_idx_scratch: TensorMut<'a, i32, 2>,
    pub values: TensorMut<'a, T, 2>,
    pub counts: TensorMut<'a, i32, 2>,
    pub counter: TensorMut<'a, i32, 1>,
}

Args bundle for a unique launch.

Fields

input: TensorRef<'a, T, 2>

Input [batch, row_len].

sorted_scratch: TensorMut<'a, T, 2>

Scratch buffer for sorted input [batch, row_len] (caller- allocated; overwritten).

sorted_idx_scratch: TensorMut<'a, i32, 2>

Scratch buffer for sorted indices [batch, row_len] (caller- allocated; overwritten — unused after the dedup).

values: TensorMut<'a, T, 2>

Output values [batch, max_unique].

counts: TensorMut<'a, i32, 2>

Output per-cell counts [batch, max_unique].

counter: TensorMut<'a, i32, 1>

Per-row counter [batch].