Use the atom transform block as the unit to count transform block
orders. This avoids the assumption that the codec goes in unit of
4x4 block.

Change-Id: I2c73c19c1349843656c940a79547a5baebbda997