It is narrated from Sayyidah Umm Khalid bint Khalid bin Sa'id bin al-As al-Akbar (may Allah be pleased with her); she said: "I heard the Messenger of Allah (peace and blessings be upon him) seeking refuge from the punishment of the grave."
Hadith Referenceالمعجم الكبير للطبراني / مسند النساء / 21708