ColossalAI/colossalai/fx/profiler/experimental/profiler_module
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