pub fn argmax_bf16(data: &[u8]) -> u32
Legacy: argmax over BF16 values (still used by argmax_on_device fallback).