Boyuan Yao 6cd784ffee [autoparallel] Add metainfo support for F.linear (#1987) ... * [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		2022-11-23 14:12:34 +08:00
..
meta_registry	[autoparallel] Add metainfo support for F.linear (#1987)	2022-11-23 14:12:34 +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 torch.nn.ReLU metainfo (#1868)	2022-11-16 23:12:31 +08:00
metainfo.py	[autoparallel] Add metainfo support for F.linear (#1987)	2022-11-23 14:12:34 +08:00
registry.py	[fx] Add linear metainfo class for auto parallel (#1783)	2022-11-04 10:55:09 +08:00