mirror of https://github.com/hpcaitech/ColossalAI
eaac03ae1d | ||
---|---|---|
.. | ||
csrc | ||
__init__.py | ||
layer_norm.py | ||
multihead_attention.py | ||
scaled_softmax.py |
eaac03ae1d | ||
---|---|---|
.. | ||
csrc | ||
__init__.py | ||
layer_norm.py | ||
multihead_attention.py | ||
scaled_softmax.py |