Taneem Ibrahim 787bc0d031 Add unit tests for pooler activation functions (#42824) ...

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

2026-05-16 14:58:16 -04:00

..

seqwise

[Perf][2/n] Eliminate GPU<->CPU syncs in pooling code (#41433)

2026-05-05 02:43:25 +00:00

tokwise

[Perf][2/n] Eliminate GPU<->CPU syncs in pooling code (#41433)

2026-05-05 02:43:25 +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

Add unit tests for pooler activation functions (#42824)

2026-05-16 14:58:16 -04: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

[Perf][2/n] Eliminate GPU<->CPU syncs in pooling code (#41433)

2026-05-05 02:43:25 +00:00