Frank Lee 262652c8bc [autoparallel] added addbmm handler (#1751)

2022-10-21 18:55:48 +08:00

..

__init__.py

[fx]patch nn.functional convolution (#1528)

2022-09-01 19:05:07 +08:00

activation_function.py

[fx] refactored the file structure of patched function and module (#1238)

2022-07-12 15:01:58 +08:00

arithmetic.py

[autoparallel] added addbmm handler (#1751)

2022-10-21 18:55:48 +08:00

convolution.py

[fx]patch nn.functional convolution (#1528)

2022-09-01 19:05:07 +08:00

embedding.py

[fx] refactored the file structure of patched function and module (#1238)

2022-07-12 15:01:58 +08:00

normalization.py

[fx] refactored the file structure of patched function and module (#1238)

2022-07-12 15:01:58 +08:00

python_ops.py

[fx]refactor tracer (#1335)

2022-07-19 15:50:42 +08:00

torch_ops.py

[fx] patched torch.max and data movement operator (#1391)

2022-08-01 15:31:50 +08:00