Function feed_argmax 

pub fn feed_argmax(row_bf16: &[u16], mask: [u32; 2]) -> u32

2026-09-25: One row of argmax_bf16_batch_feed, from the BF16 bits.