Function tropical_gemm_with_argmax_inner 

pub unsafe fn tropical_gemm_with_argmax_inner<T: TropicalWithArgmax<Index = u32>, K: MicrokernelWithArgmax<T>>(
    m: usize,
    n: usize,
    k: usize,
    a: *const T::Scalar,
    lda: usize,
    trans_a: Transpose,
    b: *const T::Scalar,
    ldb: usize,
    trans_b: Transpose,
    result: &mut GemmWithArgmax<T>,
    params: &TilingParams,
    kernel: &K,
)

Tropical GEMM with argmax tracking and custom kernel.

Safety

Same requirements as tropical_gemm_portable