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.py	[refactor] remove gpc dependency in colotensor's _ops (#1189)	2022-07-04 18:54:37 +08:00
embedding_bag.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