Make set context function aware of rectangular transform block size Account for the rectangular transform block size in setting the context data. Change-Id: Ic30a6a3eaaca4c945e0aab3acbaeb99aa48b0064