Searched refs:update_sharding (Results 1 – 3 of 3) sorted by relevance
164 const HloSharding& update_sharding, const HloInstruction& hlo);
1008 const HloSharding& update_sharding, const HloInstruction& hlo) { in ScatterOutputShardingFromUpdate() argument1027 hlo.shape(), update_sharding, inserted_window_dims, in ScatterOutputShardingFromUpdate()
2364 auto update_sharding = HloSharding::Replicate(); in HandleDynamicUpdateSlice() local2367 update_sharding = in HandleDynamicUpdateSlice()2372 GetPartitionedHlo(hlo->operand(1)).Reshard(update_sharding).hlo(); in HandleDynamicUpdateSlice()