github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/auto_parallel/solver/op_handler/__init__.py

7 lines
310 B
Python
Raw Normal View History

[autoparallel] adapt solver with resnet (#1583) * [autoparallel]adapt solver with resnet * polish code * polish code
2022-09-13 04:07:09 +00:00
from .operator_handler import OperatorHandler
from .dot_handler import DotHandler
from .conv_handler import ConvHandler
from .batch_norm_handler import BatchNormHandler
[autoparallel] add reshape handler (#1594) * [autoparallel] add reshape handler * polish code
2022-09-14 02:25:45 +00:00
from .reshape_handler import ReshapeHandler
[autoparallel] adapt solver with resnet (#1583) * [autoparallel]adapt solver with resnet * polish code * polish code
2022-09-13 04:07:09 +00:00
[autoparallel] add reshape handler (#1594) * [autoparallel] add reshape handler * polish code
2022-09-14 02:25:45 +00:00
__all__ = ['OperatorHandler', 'DotHandler', 'ConvHandler', 'BatchNormHandler', 'ReshapeHandler']