Searched refs:replicate_update (Results 1 – 1 of 1) sorted by relevance
2371 HloInstruction* replicate_update = in HandleDynamicUpdateSlice() local2374 const auto& update_shape = replicate_update->shape(); in HandleDynamicUpdateSlice()2441 partitioned_shape, partitioned_input, replicate_update, new_indices)); in HandleDynamicUpdateSlice()