Use the sharding of the appropriate tuple element rather than that of the entire tuple when computing input_shardings for get-tuple-element HLO ops with user sharding annotations.
Use the sharding of the appropriate tuple element rather than that of the entire tuple when computing input_shardings for get-tuple-element HLO ops with user sharding annotations.
