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
..
__init__.py	[autoparallel] add binary elementwise metainfo for auto parallel (#2058)	2022-12-04 15:18:51 +08:00
activation.py	[autoparallel] Patch meta information of `torch.nn.functional.softmax` and `torch.nn.Softmax` (#2674)	2023-02-13 16:09:22 +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