dayellow 204ca2f09a [NFC] polish colossalai/fx/profiler/experimental/profiler_module/embedding.py code style (#3256) ...

Co-authored-by: Minghao Huang <huangminghao@luchentech.com>

2023-03-29 15:22:21 +08:00

..

__init__.py

[fx] provide a stable but not accurate enough version of profiler. (#1547)

2022-09-07 11:21:04 +08:00

activation_function.py

[fx] provide a stable but not accurate enough version of profiler. (#1547)

2022-09-07 11:21:04 +08:00

attention.py

[fx] provide a stable but not accurate enough version of profiler. (#1547)

2022-09-07 11:21:04 +08:00

convolution.py

[doc] add deepspeed citation and copyright (#2996)

2023-03-04 20:08:11 +08:00

dropout.py

[fx] provide a stable but not accurate enough version of profiler. (#1547)

2022-09-07 11:21:04 +08:00

embedding.py

[NFC] polish colossalai/fx/profiler/experimental/profiler_module/embedding.py code style (#3256)

2023-03-29 15:22:21 +08:00

linear.py

[fx] provide a stable but not accurate enough version of profiler. (#1547)

2022-09-07 11:21:04 +08:00

normalization.py

[doc] add deepspeed citation and copyright (#2996)

2023-03-04 20:08:11 +08:00

pooling.py

[fx] provide a stable but not accurate enough version of profiler. (#1547)

2022-09-07 11:21:04 +08:00

rnn.py

[fx] provide a stable but not accurate enough version of profiler. (#1547)

2022-09-07 11:21:04 +08:00

torch_op.py

[fx] provide a stable but not accurate enough version of profiler. (#1547)

2022-09-07 11:21:04 +08:00