Boyuan Yao eae77c831d [autoparallel] Patch meta information for nodes that will not be handled by SPMD solver (#2823) ... * [autoparallel] non spmd meta information generator * [autoparallel] patch meta information for non spmd nodes		2023-02-22 10:28:56 +08:00
..
__init__.py	[autoparallel] Patch meta information for nodes that will not be handled by SPMD solver (#2823)	2023-02-22 10:28:56 +08:00
activation.py	[autoparallel] Patch meta information of `torch.tanh()` and `torch.nn.Dropout` (#2773)	2023-02-22 10:27:59 +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
embedding.py	[autoparallel] Patch meta information of `torch.nn.Embedding` (#2760)	2023-02-17 10:39:48 +08:00
linear.py	[autoparallel] Patch meta information of `torch.matmul` (#2584)	2023-02-08 11:05:31 +08:00
non_spmd.py	[autoparallel] Patch meta information for nodes that will not be handled by SPMD solver (#2823)	2023-02-22 10:28:56 +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 tensor related operations meta information (#2789)	2023-02-20 17:38:55 +08:00
tensor.py	[autoparallel] Patch tensor related operations meta information (#2789)	2023-02-20 17:38:55 +08:00
where.py	[autoparallel] Patch meta information of `torch.where` (#2822)	2023-02-22 10:28:21 +08:00