Stator cooling