Boyuan Yao 40c916b192 [autoparallel] Patch meta information of `torch.nn.functional.softmax` and `torch.nn.Softmax` (#2674) ... * [autoparallel] softmax metainfo * [autoparallel] softmax metainfo		2023-02-13 16:09:22 +08:00
..
meta_registry	[autoparallel] Patch meta information of `torch.nn.functional.softmax` and `torch.nn.Softmax` (#2674)	2023-02-13 16:09:22 +08:00
__init__.py	[fx] Add linear metainfo class for auto parallel (#1783)	2022-11-04 10:55:09 +08:00
constants.py	[autoparallel] align the data_ptr with the old version of auto activation checkpoint pipeline (#2261)	2023-01-03 10:30:15 +08:00
metainfo.py	[autoparallel] align the data_ptr with the old version of auto activation checkpoint pipeline (#2261)	2023-01-03 10:30:15 +08:00
registry.py	[fx] Add linear metainfo class for auto parallel (#1783)	2022-11-04 10:55:09 +08:00