sam21MemoryEncoder

features, [dim, grid, grid], and mask, [1, size, size] — logits, or a binary mask when binary, as a frame prompted with points is encoded — to memory, [memory.dim, grid, grid]. The host adds the occlusion embedding where the object is absent, and keeps the memory rounded to bfloat16, as the upstream keeps it; this graph reads the embedding so the file holds it.