ColossalAI/colossalai/tensor/d_tensor
Frank Lee eb39154d40
[dtensor] updated api and doc (#3845)
2023-06-08 10:18:17 +08:00
..
RAEDME.md [dtensor] updated api and doc (#3845) 2023-06-08 10:18:17 +08:00
__init__.py [dtensor] updated api and doc (#3845) 2023-06-08 10:18:17 +08:00
comm_spec.py [dtensor] updated api and doc (#3845) 2023-06-08 10:18:17 +08:00
d_tensor.py [dtensor] updated api and doc (#3845) 2023-06-08 10:18:17 +08:00
layout.py [dtensor] updated api and doc (#3845) 2023-06-08 10:18:17 +08:00
layout_converter.py [dtensor] updated api and doc (#3845) 2023-06-08 10:18:17 +08:00
misc.py [DTensor] refactor sharding spec (#2987) 2023-03-07 11:08:11 +08:00
sharding_spec.py Merge pull request #3916 from FrankLeeeee/sync/dtensor-with-develop 2023-06-07 11:50:43 +08:00
utils.py [DTensor] implement layout converter (#3055) 2023-03-10 09:53:52 +08:00