add batch size argument for crosscorr_kernel