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
..
__init__.py	[autoparallel] add pooling metainfo (#1968)	2022-11-18 15:13:03 +08:00
activation.py	[autoparallel] add torch.nn.ReLU metainfo (#1868)	2022-11-16 23:12:31 +08:00
conv.py	[autoparallel] add torch.nn.ReLU metainfo (#1868)	2022-11-16 23:12:31 +08:00
linear.py	[autoparallel] Add metainfo support for F.linear (#1987)	2022-11-23 14:12:34 +08:00
norm.py	[autoparallel] add torch.nn.ReLU metainfo (#1868)	2022-11-16 23:12:31 +08:00
pooling.py	[autoparallel] add pooling metainfo (#1968)	2022-11-18 15:13:03 +08:00