// Slice each row of `i` and `v` to perform a separate dynamic-update-slice
    // on the contents of `x`.
    auto input_type = mlir::cast<ShapedType>(input.getType());
    auto updates_type = mlir::cast<ShapedType>(updates.getType());
    auto indices_type = mlir::cast<ShapedType>(indices.getType());
    if (!input_type.hasRank()) return failure();
    if (!updates_type.hasRank() || updates_type.isDynamicDim(0))