It was clearly Drago imo. Lang was not used to anyone standing up to him and prior to the second fight Rocky was a one dimensional fighter. Appollo taught him speed and movement with which he destroyed Lang in the rematch.

He clearly showed these new skills in the fight with Drago which to me represents Balboa's finest ring performance.