xla

efb1e6fd - Also dump output sharding on HLO file (#5339)

Also dump output sharding on HLO file (#5339) * Also dump output sharding on HLO file * only dump output sharding if dump format is HLO * add test * fix typo