Skip to main content

compile_memory_encoder_ir

Function compile_memory_encoder_ir 

Source
pub fn compile_memory_encoder_ir(
    weights: &mut Sam2MemoryEncoderWeights,
    mask_in_h: usize,
    mask_in_w: usize,
    feat_h: usize,
    feat_w: usize,
    device: Device,
    profile: &CompileProfile,
) -> Result<(), Error>
Expand description

Compile memory-encoder IR subgraphs (mask down, pix 1×1, fuser, optional out 1×1).