Control the loop over transform blocks in the unit of smallest
transform block size. This would provide scalability to support
smaller transform block sizes.

Change-Id: I6a8167cf91f71c5d4af4e244c628f378866a607c