pub fn softmax_s8( input: &[i8], num_rows: usize, row_size: usize, mult: i32, shift: i32, diff_min: i32, output: &mut [i8], ) -> Result<()>
Performs Softmax for int8 tensors.