ColossalAI/colossalai/fx/tracer/meta_patch
Xue Fuzhao e070ca45c6 [NFC] polish colossalai/fx/tracer/meta_patch/patched_module/convolution.py code style (#1563) 2022-09-08 22:11:04 +08:00
..
patched_function [fx]patch nn.functional convolution (#1528) 2022-09-01 19:05:07 +08:00
patched_module [NFC] polish colossalai/fx/tracer/meta_patch/patched_module/convolution.py code style (#1563) 2022-09-08 22:11:04 +08:00
__init__.py
registry.py