YuliangLiu0306 8e4e8601b7 [DTensor] implement layout converter (#3055) ...

* [DTensor] refactor LayoutConverter for DTensor

* polish code

* polish docstring

2023-03-10 09:53:52 +08:00

..

__init__.py

[DTensor] refactor sharding spec (#2987)

2023-03-07 11:08:11 +08:00

comm_spec.py

[DTensor] refactor CommSpec (#3034)

2023-03-08 10:45:31 +08:00

d_tensor.py

[DTensor] implementation of dtensor (#2946)

2023-03-01 16:34:58 +08:00

layout_converter.py

[DTensor] implement layout converter (#3055)

2023-03-10 09:53:52 +08:00

layout.py

[DTensor] refactor sharding spec (#2987)

2023-03-07 11:08:11 +08:00

misc.py

[DTensor] refactor sharding spec (#2987)

2023-03-07 11:08:11 +08:00

sharding_spec.py

[DTensor] refactor CommSpec (#3034)

2023-03-08 10:45:31 +08:00

utils.py

[DTensor] implement layout converter (#3055)

2023-03-10 09:53:52 +08:00