Boyuan Yao cf0268da93 [autoparallel] Add F.conv metainfo (#2069) ... * [fx] metainfo class for auto parallel * [fx] add unit test for linear metainfo * [fx] fix bwd param for linear * [fx] modify unit test * [fx] modify unit test * [fx] modify import * [fx] modify import * [fx] modify import * [fx] move meta profiler to auto parallel * [fx] add conv metainfo class * [fx] restore profiler * [fx] restore meta profiler * [autoparallel] modify unit test * [fx] modify unit test * [autoparallel] add batchnorm metainfo class * [autoparallel] fix batchnorm unit test function declaration * [fx] restore profiler * [fx] add relu metainfo class * [fx] restore profiler * [autoparallel] modify metainfo input * [autoparallel] add pooling metainfo * [autoparallel] add F.linear metainfo generator * [autoparallel] add binary elementwise metainfo * [fx] recover profiler * [autoparallel] fix forward memory calculation * [autoparallel] modify constants.py * [autoparallel] remove redundant print * [autoparallel] add F.conv metainfo * [autoparallel] linear fix		2022-12-06 10:17:57 +08:00
..
meta_registry	[autoparallel] Add F.conv metainfo (#2069)	2022-12-06 10:17:57 +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] add binary elementwise metainfo for auto parallel (#2058)	2022-12-04 15:18:51 +08:00
registry.py	[fx] Add linear metainfo class for auto parallel (#1783)	2022-11-04 10:55:09 +08:00