Boyuan Yao 0385b26ebf [autoparallel] Patch meta information of `torch.nn.LayerNorm` (#2647) ... * [autoparallel] layernorm metainfo patch * [autoparallel] polish test		2023-02-10 14:29:24 +08:00
..
__init__.py	[autoparallel] add binary elementwise metainfo for auto parallel (#2058)	2022-12-04 15:18:51 +08:00
activation.py	[autoparallel] patch torch.flatten metainfo for autoparallel (#2247)	2023-01-02 15:51:03 +08:00
binary_elementwise_ops.py	[autoparallel] bypass MetaInfo when unavailable and modify BCAST_FUNC_OP metainfo (#2293)	2023-01-03 20:28:01 +08:00
conv.py	[autoparallel] Attach input, buffer and output tensor to MetaInfo class (#2162)	2022-12-28 13:37:40 +08:00
linear.py	[autoparallel] Patch meta information of `torch.matmul` (#2584)	2023-02-08 11:05:31 +08:00
norm.py	[autoparallel] Patch meta information of `torch.nn.LayerNorm` (#2647)	2023-02-10 14:29:24 +08:00
pooling.py	[autoparallel] patch torch.flatten metainfo for autoparallel (#2247)	2023-01-02 15:51:03 +08:00