func::FuncOp ResolveCondFunction(::mlir::SymbolTableCollection* table) {
      if (table)
        return table->lookupNearestSymbolFrom<func::FuncOp>(*this, getCondAttr());
      return SymbolTable::lookupNearestSymbolFrom<func::FuncOp>(*this, getCondAttr());
    }
    // TODO(b/204997177): Deprecate and remove.
    func::FuncOp cond_function() { return ResolveCondFunction(nullptr); }

    // Get the body function.

          module, compatible_types,
          {while_op.ResolveCondFunction(&symbol_table_),
           while_op.ResolveBodyFunction(&symbol_table_)},
          max_iterations);
    }
    return PropagateShapeToFunctions(
        module, while_op.getInput().getTypes(),
        {while_op.ResolveCondFunction(&symbol_table_),
         while_op.ResolveBodyFunction(&symbol_table_)},