xla

8b8d2898 - Initialize logical_element_type before sharding in XLATensor::Data (#4418)