Shift symmetries in (anti) de Sitter space