Handle InputShape type shaping too.