Taneem Ibrahim 9464529612 [ModelRunner v2] Enable decoder token-wise pooling (#50931) ...

Signed-off-by: Taneem Ibrahim <taneem.ibrahim@gmail.com>

2026-08-06 22:21:52 +00:00

..

seqwise

[Core] Ensure memory is pinned prior to async h2d copy (#45424)

2026-06-20 20:02:24 -07:00

tokwise

[ModelRunner v2] Enable decoder token-wise pooling (#50931)

2026-08-06 22:21:52 +00:00

__init__.py

[Model] Reorganize pooling layers (#31973)

2026-01-09 11:02:14 +00:00

abstract.py

[Model] Reorganize pooling layers (#31973)

2026-01-09 11:02:14 +00:00

activations.py

Added extra_repr() to pooler classes to improve debuggability (#44805)

2026-06-08 03:19:31 +00:00

common.py

[Perf] Remove redundant device copies for CPU-only pooling token IDs, 48.9% E2E throughput improvement (#38139)

2026-03-29 18:12:50 +00:00

special.py

Added extra_repr() to pooler classes to improve debuggability (#44805)

2026-06-08 03:19:31 +00:00