mirror of https://github.com/hpcaitech/ColossalAI
7531c6271f
* [fx] refactored the file structure of patched function and module * polish code |
||
---|---|---|
.. | ||
__init__.py | ||
activation_function.py | ||
arithmetic.py | ||
embedding.py | ||
normalization.py | ||
python_ops.py | ||
torch_ops.py |