from dinoreg.modules.kpconv.kpconv import KPConv
from dinoreg.modules.kpconv.modules import (
    ConvBlock,
    ResidualBlock,
    UnaryBlock,
    LastUnaryBlock,
    GroupNorm,
    KNNInterpolate,
    GlobalAvgPool,
    MaxPool,
)
from dinoreg.modules.kpconv.functional import nearest_upsample, global_avgpool, maxpool
