Struct Sam2MemoryMaskDownCompiled 

pub struct Sam2MemoryMaskDownCompiled {
    pub embed_dim: usize,
    pub in_h: usize,
    pub in_w: usize,
    pub out_h: usize,
    pub out_w: usize,
    /* private fields */
}

Fields

embed_dim: usize

in_h: usize

in_w: usize

out_h: usize

out_w: usize

Implementations

impl Sam2MemoryMaskDownCompiled

pub fn compile( w: &Sam2MaskDownSamplerWeights, in_h: usize, in_w: usize, device: Device, ) -> Result<Sam2MemoryMaskDownCompiled, Error>

pub fn compile_with_profile( w: &Sam2MaskDownSamplerWeights, in_h: usize, in_w: usize, device: Device, profile: &CompileProfile, ) -> Result<Sam2MemoryMaskDownCompiled, Error>

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

Flat [1, H, W] mask (post-sigmoid).