Improve performance of layout assignment

Previously creating the layout constraints was proportional to the
number of buffers multiplied by the number of computations what is bad
when both of them is big.

The new implementation is proportional with the total number of
instructions by calculating the effected buffers from the tuple point
sets.

PiperOrigin-RevId: 204270924