IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /get-started.md). For the complete documentation index, see llms.txt.

Skip to main content

For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo module

spatial_merge

Implements spatial merge, which compresses vision token grids by merging spatial blocks before attention.

Functions

spatial_merge: Launches the spatial merge kernel that compresses vision token grids by merging spatial blocks before attention.
spatial_merge_kernel: Spatial merge kernel.

Functions