support partition the convnet