2022-09-26 08:58:14 +00:00
|
|
|
from .strategy_generator import StrategyGenerator_V2
|
|
|
|
|
from .matmul_strategy_generator import DotProductStrategyGenerator, MatVecStrategyGenerator, LinearProjectionStrategyGenerator, BatchedMatMulStrategyGenerator
|
2022-09-28 03:24:59 +00:00
|
|
|
from .conv_strategy_generator import ConvStrategyGenerator
|
2022-09-29 04:49:28 +00:00
|
|
|
from .batch_norm_generator import BatchNormStrategyGenerator
|
2022-10-09 09:30:42 +00:00
|
|
|
from .unary_elementwise_generator import UnaryElementwiseGenerator
|
2022-10-09 06:49:18 +00:00
|
|
|
from .getitem_generator import GetItemStrategyGenerator, TensorStrategyGenerator, TensorTupleStrategyGenerator
|
2022-10-09 06:23:22 +00:00
|
|
|
from .layer_norm_generator import LayerNormGenerator
|
2022-09-26 08:58:14 +00:00
|
|
|
|
|
|
|
|
__all__ = [
|
|
|
|
|
'StrategyGenerator_V2', 'DotProductStrategyGenerator', 'MatVecStrategyGenerator',
|
2022-10-09 09:30:42 +00:00
|
|
|
'LinearProjectionStrategyGenerator', 'BatchedMatMulStrategyGenerator', 'ConvStrategyGenerator', 'UnaryElementwiseGenerator',
|
2022-10-09 06:49:18 +00:00
|
|
|
'BatchNormStrategyGenerator', 'GetItemStrategyGenerator', 'TensorStrategyGenerator', 'TensorTupleStrategyGenerator',
|
|
|
|
|
'LayerNormGenerator'
|
2022-09-26 08:58:14 +00:00
|
|
|
]
|
