Skip to main content

softmax_s8

Function softmax_s8 

Source
pub fn softmax_s8(
    input: &[i8],
    num_rows: usize,
    row_size: usize,
    mult: i32,
    shift: i32,
    diff_min: i32,
    output: &mut [i8],
) -> Result<()>
Expand description

Performs Softmax for int8 tensors.