Jiarui Fang 363fc2861a [embeddings] more detailed timer (#1692)

2022-10-12 12:01:21 +08:00

..

__init__.py

[embedding] polish parallel embedding tablewise (#1545)

2022-09-06 10:41:20 +08:00

base_embedding.py

[FAW] export FAW in _ops (#1438)

2022-08-11 13:43:24 +08:00

cache_mgr.py

[embeddings] more detailed timer (#1692)

2022-10-12 12:01:21 +08:00

copyer.py

[NFC] polish doc style for ColoTensor (#1457)

2022-08-16 09:21:05 +08:00

embedding_config.py

[embedding] polish parallel embedding tablewise (#1545)

2022-09-06 10:41:20 +08:00

freq_aware_embedding.py

[embedding] polish async copy (#1657)

2022-09-27 14:37:03 +08:00

parallel_freq_aware_embedding_tablewise_split_cache.py

[embedding] polish parallel embedding tablewise (#1545)

2022-09-06 10:41:20 +08:00

parallel_freq_aware_embedding_tablewise.py

[embedding] isolate cache_op from forward (#1645)

2022-09-26 11:18:59 +08:00

parallel_freq_aware_embedding.py

[embedding] polish async copy (#1657)

2022-09-27 14:37:03 +08:00