Boyuan Yao c8c79102f0 [autoparallel] patch torch.flatten metainfo for autoparallel (#2247) ...

* [autoparallel] patch torch.flatten

2023-01-02 15:51:03 +08:00

..

meta_registry

[autoparallel] patch torch.flatten metainfo for autoparallel (#2247)

2023-01-02 15:51:03 +08:00

__init__.py

[fx] Add linear metainfo class for auto parallel (#1783)

2022-11-04 10:55:09 +08:00

constants.py

[autoparallel] add binary elementwise metainfo for auto parallel (#2058)

2022-12-04 15:18:51 +08:00

metainfo.py

[autoparallel] Attach input, buffer and output tensor to MetaInfo class (#2162)

2022-12-28 13:37:40 +08:00

registry.py

[fx] Add linear metainfo class for auto parallel (#1783)

2022-11-04 10:55:09 +08:00