This website requires JavaScript.
Explore
Help
Register
Sign In
github
/
ColossalAI
Watch
1
Star
0
Fork
0
You've already forked ColossalAI
mirror of
https://github.com/hpcaitech/ColossalAI.git
synced
2026-01-19 17:16:39 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
4df6471f5d044150edfedfe1cf8370a604e096ca
ColossalAI
/
colossalai
/
tensor
/
_ops
History
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