| 56 | } // namespace tensorflow |
| 57 | |
| 58 | int main(int argc, char* argv[]) { |
| 59 | tensorflow::port::InitMain(argv[0], &argc, &argv); |
| 60 | if (argc != 5) { |
| 61 | for (int i = 1; i < argc; ++i) { |
| 62 | fprintf(stderr, "Arg %d = %s\n", i, argv[i]); |
| 63 | } |
| 64 | fprintf(stderr, |
| 65 | "Usage: %s out.h out.cc include_internal " |
| 66 | "api_def_dirs1,api_def_dir2 ...\n" |
| 67 | " include_internal: 1 means include internal ops\n", |
| 68 | argv[0]); |
| 69 | exit(1); |
| 70 | } |
| 71 | |
| 72 | bool include_internal = tensorflow::StringPiece("1") == argv[3]; |
| 73 | std::vector<tensorflow::string> api_def_dirs = tensorflow::str_util::Split( |
| 74 | argv[4], ",", tensorflow::str_util::SkipEmpty()); |
| 75 | tensorflow::PrintAllCCOps(argv[1], argv[2], include_internal, api_def_dirs); |
| 76 | return 0; |
| 77 | } |
nothing calls this directly
no test coverage detected