ColossalAI/colossalai/tensor
Ziyue Jiang 26d4ab8b03
[Tensor] Add function to spec and update linear 1Drow and unit tests (#869)
2022-04-26 10:15:26 +08:00
..
_ops [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
colo_tensor.py [Tensor] Add function to spec and update linear 1Drow and unit tests (#869) 2022-04-26 10:15:26 +08:00
op_wrapper.py [tensor] reorganize files (#820) 2022-04-21 14:15:48 +08:00
spec.py [Tensor] Add function to spec and update linear 1Drow and unit tests (#869) 2022-04-26 10:15:26 +08:00
utils.py [tensor] reorganize files (#820) 2022-04-21 14:15:48 +08:00