Mojo module
topk_fi
Structs
-
ValueCount: A struct that holds a value and a count, used for block reductions.
Functions
-
device_sampling_from_prob: Device-level sampling from probability distribution with atomic operations. -
get_min_max_value: Compute the minimum and maximum values from input data using block reduction. -
topk_mask_logits: -
topk_sampling_from_prob: Top-K sampling from probability distribution. -
TopKMaskLogitsKernel: -
TopKSamplingFromProbKernel: Kernel for top-k sampling from probability distribution.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!