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