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