Skip to main content

rms_norm_device

Function rms_norm_device 

Source
pub fn rms_norm_device(
    x: impl AsRef<Array>,
    weight: impl AsRef<Array>,
    eps: f32,
    stream: impl AsRef<Stream>,
) -> Result<Array>
👎Deprecated since 0.26.0:

use with_stream or with_device around rms_norm

Expand description

Compatibility shim for rms_norm.