The node you are looking for is at set-reshape-callback.