forked from ccf-ai-infra/TileOPs-Metax
22 lines
741 B
Python
22 lines
741 B
Python
from .avg_pool1d import AvgPool1dKernel, AvgPool1dSpatialKernel
|
|
from .avg_pool2d import AvgPool2dKernel, AvgPool2dSpatialKernel
|
|
from .avg_pool3d import AvgPool3dKernel, AvgPool3dSpatialKernel
|
|
from .max_pool1d import MaxPool1dKernel, MaxPool1dWithIndicesKernel
|
|
from .max_pool2d import MaxPool2dKernel, MaxPool2dWithIndicesKernel
|
|
from .max_pool3d import MaxPool3dKernel, MaxPool3dWithIndicesKernel
|
|
|
|
__all__ = [
|
|
"AvgPool1dKernel",
|
|
"AvgPool1dSpatialKernel",
|
|
"AvgPool2dKernel",
|
|
"AvgPool2dSpatialKernel",
|
|
"AvgPool3dKernel",
|
|
"AvgPool3dSpatialKernel",
|
|
"MaxPool1dKernel",
|
|
"MaxPool1dWithIndicesKernel",
|
|
"MaxPool2dKernel",
|
|
"MaxPool2dWithIndicesKernel",
|
|
"MaxPool3dKernel",
|
|
"MaxPool3dWithIndicesKernel",
|
|
]
|