auto update_computation = symbol_table.lookup<mlir::func::FuncOp>(
        scatter_op->getAttr(kScatterRegionFuncName)
            .cast<mlir::FlatSymbolRefAttr>()
            .getValue());
    InlineVhloOpRegion(scatter_op.getUpdateComputation(), update_computation);
    scatter_op->removeAttr(kScatterRegionFuncName);
    to_delete_funcs.push_back(update_computation);
  });