ColossalAI/colossalai/tensor
Jiarui Fang 1190b2c4a4
[tensor] add cross_entrophy_loss (#868)
2022-04-25 16:01:52 +08:00
..
_ops [tensor] add cross_entrophy_loss (#868) 2022-04-25 16:01:52 +08:00
__init__.py [tensor] reorganize files (#820) 2022-04-21 14:15:48 +08:00
colo_tensor.py colo init context add device attr. (#866) 2022-04-25 14:24:26 +08:00
op_wrapper.py [tensor] reorganize files (#820) 2022-04-21 14:15:48 +08:00
spec.py [tensor] an initial dea of tensor spec (#865) 2022-04-25 13:33:52 +08:00
utils.py [tensor] reorganize files (#820) 2022-04-21 14:15:48 +08:00