rewrite stereoJoin_updateGradInput