Function PrintAllCCOps

tensorflow/cc/framework/cc_op_gen_main.cc:30–53

namespace {

void PrintAllCCOps(const std::string& dot_h, const std::string& dot_cc,
                   bool include_internal,
                   const std::vector<string>& api_def_dirs) {
  OpList ops;
  OpRegistry::Global()->Export(include_internal, &ops);
  ApiDefMap api_def_map(ops);
  if (!api_def_dirs.empty()) {
    Env* env = Env::Default();
    // Only load files that correspond to "ops".
    for (const auto& op : ops.op()) {
      for (const auto& api_def_dir : api_def_dirs) {
        const std::string api_def_file_pattern =
            io::JoinPath(api_def_dir, "api_def_" + op.name() + ".pbtxt");
        if (env->FileExists(api_def_file_pattern).ok()) {
          TF_CHECK_OK(api_def_map.LoadFile(env, api_def_file_pattern));
        }
      }
    }
  }

  api_def_map.UpdateDocs();

  WriteCCOps(ops, api_def_map, dot_h, dot_cc);
}

}  // namespace
}  // namespace tensorflow