pub fn upsample_weights( full_depth: f32, depths: [f32; 4], fractions: [f32; 2], depth_scale: f32, ) -> [f32; 4]