[3DConvFFT] Fix the 3d conv layer setup