Jiarui Fang 909211453b [Tensor] Add some attributes to ColoTensor (#877) ...

* [Tensor] add some function to ColoTensor

* torch.allclose

* rm torch.add

2022-04-26 15:10:47 +08:00

..

__init__.py

[tensor] customized op returns ColoTensor (#875)

2022-04-26 13:23:59 +08:00

element_wise.py

[Tensor] Add some attributes to ColoTensor (#877)

2022-04-26 15:10:47 +08:00

layernorm.py

[tensor] add cross_entrophy_loss (#868)

2022-04-25 16:01:52 +08:00

linear.py

[tensor] customized op returns ColoTensor (#875)

2022-04-26 13:23:59 +08:00

loss.py

[tensor] add cross_entrophy_loss (#868)

2022-04-25 16:01:52 +08:00