Struct Sam2MemoryPrefixCompiled 

pub struct Sam2MemoryPrefixCompiled {
    pub in_dim: usize,
    pub mask_in_h: usize,
    pub mask_in_w: usize,
    pub feat_h: usize,
    pub feat_w: usize,
    /* private fields */
}

Fused mask down + pix_feat_proj + add → fuser input.

Fields

in_dim: usize

mask_in_h: usize

mask_in_w: usize

feat_h: usize

feat_w: usize

Implementations

impl Sam2MemoryPrefixCompiled

pub fn compile( mask_ds: &Sam2MaskDownSamplerWeights, in_dim: usize, mask_in_h: usize, mask_in_w: usize, feat_h: usize, feat_w: usize, pix_w: &[f32], pix_b: &[f32], device: Device, ) -> Result<Sam2MemoryPrefixCompiled, Error>

pub fn compile_with_profile( mask_ds: &Sam2MaskDownSamplerWeights, in_dim: usize, mask_in_h: usize, mask_in_w: usize, feat_h: usize, feat_w: usize, pix_w: &[f32], pix_b: &[f32], device: Device, profile: &CompileProfile, ) -> Result<Sam2MemoryPrefixCompiled, Error>

pub fn run(&mut self, mask: &[f32], pix_feat: &[f32]) -> Result<Vec<f32>, Error>

Post-sigmoid mask [1, H, W] flat + pix_feat NCHW [in_dim, feat_h, feat_w].