| 66 | } |
| 67 | |
| 68 | int Run(int argc, char** argv) { |
| 69 | string FLAGS_profile_path = ""; |
| 70 | string FLAGS_graph_path = ""; |
| 71 | string FLAGS_run_meta_path = ""; |
| 72 | string FLAGS_op_log_path = ""; |
| 73 | string FLAGS_checkpoint_path = ""; |
| 74 | int32 FLAGS_max_depth = 10; |
| 75 | int64 FLAGS_min_bytes = 0; |
| 76 | int64 FLAGS_min_peak_bytes = 0; |
| 77 | int64 FLAGS_min_residual_bytes = 0; |
| 78 | int64 FLAGS_min_output_bytes = 0; |
| 79 | int64 FLAGS_min_micros = 0; |
| 80 | int64 FLAGS_min_accelerator_micros = 0; |
| 81 | int64 FLAGS_min_cpu_micros = 0; |
| 82 | int64 FLAGS_min_params = 0; |
| 83 | int64 FLAGS_min_float_ops = 0; |
| 84 | int64 FLAGS_min_occurrence = 0; |
| 85 | int64 FLAGS_step = -1; |
| 86 | string FLAGS_order_by = "name"; |
| 87 | string FLAGS_account_type_regexes = ".*"; |
| 88 | string FLAGS_start_name_regexes = ".*"; |
| 89 | string FLAGS_trim_name_regexes = ""; |
| 90 | string FLAGS_show_name_regexes = ".*"; |
| 91 | string FLAGS_hide_name_regexes; |
| 92 | bool FLAGS_account_displayed_op_only = false; |
| 93 | string FLAGS_select = "micros"; |
| 94 | string FLAGS_output = ""; |
| 95 | for (int i = 0; i < argc; i++) { |
| 96 | fprintf(stderr, "%s\n", argv[i]); |
| 97 | } |
| 98 | |
| 99 | std::vector<Flag> flag_list = { |
| 100 | Flag("profile_path", &FLAGS_profile_path, "Profile binary file name."), |
| 101 | Flag("graph_path", &FLAGS_graph_path, "GraphDef proto text file name"), |
| 102 | Flag("run_meta_path", &FLAGS_run_meta_path, |
| 103 | "Comma-separated list of RunMetadata proto binary " |
| 104 | "files. Each file is given step number 0,1,2,etc"), |
| 105 | Flag("op_log_path", &FLAGS_op_log_path, |
| 106 | "tensorflow::tfprof::OpLogProto proto binary file name"), |
| 107 | Flag("checkpoint_path", &FLAGS_checkpoint_path, |
| 108 | "TensorFlow Checkpoint file name"), |
| 109 | Flag("max_depth", &FLAGS_max_depth, "max depth"), |
| 110 | Flag("min_bytes", &FLAGS_min_bytes, "min_bytes"), |
| 111 | Flag("min_peak_bytes", &FLAGS_min_peak_bytes, "min_peak_bytes"), |
| 112 | Flag("min_residual_bytes", &FLAGS_min_residual_bytes, |
| 113 | "min_residual_bytes"), |
| 114 | Flag("min_output_bytes", &FLAGS_min_output_bytes, "min_output_bytes"), |
| 115 | Flag("min_micros", &FLAGS_min_micros, "min micros"), |
| 116 | Flag("min_accelerator_micros", &FLAGS_min_accelerator_micros, |
| 117 | "min accelerator_micros"), |
| 118 | Flag("min_cpu_micros", &FLAGS_min_cpu_micros, "min_cpu_micros"), |
| 119 | Flag("min_params", &FLAGS_min_params, "min params"), |
| 120 | Flag("min_float_ops", &FLAGS_min_float_ops, "min float ops"), |
| 121 | Flag("min_occurrence", &FLAGS_min_occurrence, "min occurrence"), |
| 122 | Flag("step", &FLAGS_step, |
| 123 | "The stats of which step to use. By default average"), |
| 124 | Flag("order_by", &FLAGS_order_by, "order by"), |
| 125 | Flag("account_type_regexes", &FLAGS_start_name_regexes, |