PR #13477: Add SetupDerivedInstruction for whileop in spmd partitioner
PR #13477: Add SetupDerivedInstruction for whileop in spmd partitioner

Imported from GitHub PR https://github.com/openxla/xla/pull/13477

This is a minor change. It calls the SetupDerivedInstruction for while operation to make sure attributes are propagated properly.
Copybara import of the project:

--
9563c00004bc2a988d01964f213e5dc4dc535489 by Shraiysh Vaishay <svaishay@nvidia.com>:

Add SetupDerivedInstruction for whileop in spmd partitioner

Merging this change closes #13477

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/13477 from shraiysh:fix-while-loop-derived-instruction 9563c00004bc2a988d01964f213e5dc4dc535489
