// The main_outputs may include tokens that are not among the op_results;
  mlir::TypeRange main_output_types = loader->OutputTypes();
  int nr_main_token_outputs =
      llvm::count_if(main_output_types, tensorflow::IsTokenType);
  if (op_results.size() != main_output_types.size() - nr_main_token_outputs) {
    llvm::errs() << "XlaCallModule has " << op_results.size()