In the publication, only the last layer uses U and V, but in the code it seems that every layers used U and V