Amit Moryossef d518e89573 cache position grids in NaViT forward pass (#354) ...

Use lru_cache to cache unique (ph, pw, device) position grids, avoiding
redundant computation when multiple images share the same patch
dimensions. Cache persists across forward passes.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-authored-by: Claude Opus 4.5 <noreply@anthropic.com>

2025-12-07 04:32:30 -08:00

13 KiB

Raw Blame History

View Raw