Searched defs:data_sharding (Results 1 – 1 of 1) sorted by relevance
730 HloSharding ScatterIndexSharding(const HloSharding& data_sharding, in ScatterIndexSharding()859 HloSharding ScatterEffectiveDataSharding(const HloSharding& data_sharding, in ScatterEffectiveDataSharding()