// the given type.
TensorType UpdateElementTypeTo(Type ty, Type element_ty) {
  auto ranked_ty = mlir::dyn_cast<RankedTensorType>(ty);
  if (!ranked_ty) {
    return UnrankedTensorType::get(element_ty);
  }
  return RankedTensorType::get(ranked_ty.getShape(), element_ty,
                               ranked_ty.getEncoding());
}

template <typename SrcOpT, typename DstOpT>