Jiarui Fang 1b657f9ce1 [tensor] revert local view back (#1178)

2022-06-27 18:38:34 +08:00

..

__init__.py

[tensor] add embedding bag op (#1156)

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

_utils.py

reorgnize colotensor directory (#1062)

2022-06-03 18:04:22 +08:00

addmm.py

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

2022-06-27 15:56:11 +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

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

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

embedding.py

[tensor] revert local view back (#1178)

2022-06-27 18:38:34 +08:00

layernorm.py

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

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

linear.py

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

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

loss.py

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

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