ColossalAI/colossalai/tensor/_ops
Ziyue Jiang 26d4ab8b03
[Tensor] Add function to spec and update linear 1Drow and unit tests (#869)
2022-04-26 10:15:26 +08:00
..
__init__.py [Tensor] Add function to spec and update linear 1Drow and unit tests (#869) 2022-04-26 10:15:26 +08:00
element_wise.py [Tensor] add layer norm Op (#852) 2022-04-25 11:49:20 +08:00
init.py [tensor] reorganize files (#820) 2022-04-21 14:15:48 +08:00
layernorm.py [tensor] add cross_entrophy_loss (#868) 2022-04-25 16:01:52 +08:00
linear.py [Tensor] Add function to spec and update linear 1Drow and unit tests (#869) 2022-04-26 10:15:26 +08:00
loss.py [tensor] add cross_entrophy_loss (#868) 2022-04-25 16:01:52 +08:00