Jiarui Fang 521078ffc9 [embedding] fix a bug in table wise sharding (#1538)

2022-09-02 15:48:35 +08:00

..

__init__.py

[embedding] add tablewise sharding for FAW (#1526)

2022-09-01 17:55:41 +08:00

base_embedding.py

[FAW] export FAW in _ops (#1438)

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

cache_mgr.py

[doc] docstring for FreqAwareEmbeddingBag (#1525)

2022-08-31 13:52:30 +08:00

copyer.py

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

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

freq_aware_embedding.py

[embedding] add tablewise sharding for FAW (#1526)

2022-09-01 17:55:41 +08:00

parallel_freq_aware_embedding_tablewise.py

[embedding] fix a bug in table wise sharding (#1538)

2022-09-02 15:48:35 +08:00

parallel_freq_aware_embedding.py

[embedding] add tablewise sharding for FAW (#1526)

2022-09-01 17:55:41 +08:00