Searched defs:update_sharding (Results 1 – 2 of 2) sorted by relevance
1008 const HloSharding& update_sharding, const HloInstruction& hlo) { in ScatterOutputShardingFromUpdate()
2364 auto update_sharding = HloSharding::Replicate(); in HandleDynamicUpdateSlice() local