<< func_arg_count;
  }

  return success();
}

void LegacyCallOp::setCalleeFromCallable(mlir::CallInterfaceCallable callee) {
  // Direct call.
  if (SymbolRefAttr fAttr = getFAttr()) {
    SymbolRefAttr calleeAttr = callee.get<SymbolRefAttr>();
    return setFAttr(cast<FlatSymbolRefAttr>(calleeAttr));
  }
  // Indirect call, callee Value is the first operand.

}

template <typename CallOpClass>
static void SetPartitionCalleeFromCallable(CallOpClass op,
                                           mlir::CallInterfaceCallable callee) {
  // Direct call.
  if (SymbolRefAttr fAttr = op.getFAttr()) {
    SymbolRefAttr calleeAttr = callee.get<SymbolRefAttr>();
    return op.setFAttr(cast<FlatSymbolRefAttr>(calleeAttr));
  }
  // Indirect call, callee Value is the first operand.