Remove num_layers_in_block param from UniTransformer.
Remove num_layers_in_block param from UniTransformer.

PiperOrigin-RevId: 364680811

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo