Jiarui Fang 060b917daf [refactor] remove gpc dependency in colotensor's _ops (#1189)

2022-07-04 18:54:37 +08:00

..

__init__.py

[tensor] add embedding bag op (#1156)

2022-06-22 15:54:03 +08:00

_utils.py

[refactor] remove gpc dependency in colotensor's _ops (#1189)

2022-07-04 18:54:37 +08:00

addmm.py

[refactor] remove gpc dependency in colotensor's _ops (#1189)

2022-07-04 18:54:37 +08:00

element_wise.py

[ColoTensor] rename APIs and add output_replicate to ComputeSpec (#1168)

2022-06-24 13:08:54 +08:00

embedding_bag.py

[refactor] remove gpc dependency in colotensor's _ops (#1189)

2022-07-04 18:54:37 +08:00

embedding.py

[refactor] remove gpc dependency in colotensor's _ops (#1189)

2022-07-04 18:54:37 +08:00

layernorm.py

[refactor] remove gpc dependency in colotensor's _ops (#1189)

2022-07-04 18:54:37 +08:00

linear.py

[refactor] remove gpc dependency in colotensor's _ops (#1189)

2022-07-04 18:54:37 +08:00

loss.py

[Tensor] rename some APIs in TensorSpec and Polish view unittest (#1176)

2022-06-27 15:56:11 +08:00