pub fn rms_norm_weight_offset( input: &[f32], weight: &[f32], rows: usize, cols: usize, eps: f32, weight_offset: f32, ) -> Vec<f32>