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