Searched defs:index_sharding (Results 1 – 1 of 1) sorted by relevance
499 HloSharding GatherOutputSharding(const HloSharding& index_sharding, in GatherOutputSharding()672 HloSharding ScatterDataSharding(const HloSharding& index_sharding, in ScatterDataSharding()706 HloSharding ScatterEffectiveIndexSharding(const HloSharding& index_sharding, in ScatterEffectiveIndexSharding()