Yōga Sūtras
Yōga Sūtra 1.38
स्वप्ननिद्राज्ञानालम्बनं वा।
(Such Chitta) supports the awareness of dream and sleep state as well.
Once a Yogi reaches this state, he/she will not lose the awareness even in sleep and dream state.