mirror of https://github.com/hpcaitech/ColossalAI
7d6293927f
* [fx] patched torch.max and data movement operator * polish code |
||
---|---|---|
.. | ||
__init__.py | ||
activation_function.py | ||
arithmetic.py | ||
embedding.py | ||
normalization.py | ||
python_ops.py | ||
torch_ops.py |