Mojo module
topk
Aliases
-
SEED = 0
:
Structs
Functions
-
bottom_k_shape
: -
top_k
: Implementation of the Top K algorithm. Returns the top or bottom K elements and their index along a specified axis. -
top_k_fused_sampling_cpu
: Generalized implementation of the Top K algorithm with sampling. Returns the sampled index from the innermost dimension of the input tensor for each row/subvolume. -
top_k_shape
: -
top_k_shape_impl
: Compute the output shape of a top/bottom k operation. -
topk_fused_sampling_gpu
: Top K algorithm with fused sampling. Returns the sampled indices from the Top-K of the innermost dimension of the input tensor for each row/subvolume. -
topk_gpu
: Generalized implementation of the Top K algorithm with/without sampling. Returns the sampled index from the innermost dimension of the input tensor for each row/subvolume or the top K values and indices across the tensor.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!