//
    // will be transformed to
    //
    // func.func @f(%x : bar) {
    //   ...
    // }
    // ...
    // func.call @f(%x : bar)

    CallInterfaceCallable callable = call.getCallableForCallee();
    mlir::SymbolRefAttr sym = callable.dyn_cast<mlir::SymbolRefAttr>();
    auto symbol =
        mlir::SymbolTable::lookupNearestSymbolFrom(call.getOperation(), sym);
    if (!symbol) return failure();

    MutableOperandRange getArgOperandsMutable() {
      return getArgsMutable();
    }

    // Return the callee of this operation.
    CallInterfaceCallable getCallableForCallee() { return getCalleeAttr(); }
    // Sets the callee from the callable
    void setCalleeFromCallable(CallInterfaceCallable callee);
  }];

  let assemblyFormat = [{

  }
  for (auto [ret, operand] :
       llvm::zip(call.getResults(), yield.getOperands())) {
    if (ret != operand) return std::nullopt;
  }
  SymbolRefAttr symbol = call.getCallableForCallee().get<SymbolRefAttr>();
  if (!symbol) return std::nullopt;
  return symbol.getLeafReference();
}

// Extracts the contents of a region with a single block into a new function.

                           StringAttr old_group) {
  // Adds the TPUReplicateMetatdataOp and TPUCompilationResultOp ops to the
  // function called by the provided 'caller'.
  mlir::CallInterfaceCallable callable = caller.getCallableForCallee();
  mlir::SymbolRefAttr sym = callable.dyn_cast<mlir::SymbolRefAttr>();
  auto func = llvm::dyn_cast<mlir::func::FuncOp>(
      mlir::SymbolTable::lookupNearestSymbolFrom(caller, sym));