[ttx/npu] Optimize top_k_sampling by replacing argsort with iterative extraction#318

Neuromancer42 merged 1 commit into

XPU-Forces:masterXPU-Forces/mojo_opset:masterfrom

lyujheng:lvzheng/ttx-optimize-top-k-samplinglyujheng/mojo_opset:lvzheng/ttx-optimize-top-k-samplingCopy head branch name to clipboard