Calculation of heat transfer in the elements of a centrifugal compression stage with internal cooling