Searched defs:update_sharding (Results 1 – 2 of 2) sorted by relevance
1134 const HloSharding& update_sharding, const HloScatterInstruction& scatter) { in ScatterOutputShardingFromUpdate()
3108 auto update_sharding = HloSharding::Replicate(); in HandleDynamicUpdateSlice() local