fn layer_norm_internal_type()

in src/lib.rs [11:19]

• 9 lines of code
• 1 McCabe index (conditional complexity)

fn layer_norm_internal_type(dtype: DType) -> Result<u32> {
    let internal_type = match dtype {
        DType::F16 => 0,
        DType::BF16 => 1,
        DType::F32 => 2,
        dtype => candle::bail!("dtype {dtype:?} is not supported"),
    };
    Ok(internal_type)
}