in tensorflow/tensorflow/core/profiler/internal/tfprof_utils.cc [115:290]
tensorflow::Status ParseCmdLine(const string& line, string* cmd,
tensorflow::tfprof::Options* opts) {
std::vector<string> pieces =
str_util::Split(line, ' ', str_util::SkipEmpty());
std::vector<string> cmds_str(kCmds, kCmds + sizeof(kCmds) / sizeof(*kCmds));
if (std::find(cmds_str.begin(), cmds_str.end(), pieces[0]) ==
cmds_str.end()) {
return tensorflow::Status(tensorflow::error::INVALID_ARGUMENT,
"First string must be a valid command.");
}
*cmd = pieces[0];
for (int i = 1; i < pieces.size(); ++i) {
if (pieces[i] == string(tensorflow::tfprof::kOptions[0])) {
if (pieces.size() <= i + 1 ||
!strings::safe_strto32(pieces[i + 1], &opts->max_depth)) {
return ReturnError(pieces, i);
}
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[1]) {
if (pieces.size() <= i + 1 ||
!strings::safe_strto64(pieces[i + 1], &opts->min_bytes)) {
return ReturnError(pieces, i);
}
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[2]) {
if (pieces.size() <= i + 1 ||
!strings::safe_strto64(pieces[i + 1], &opts->min_peak_bytes)) {
return ReturnError(pieces, i);
}
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[3]) {
if (pieces.size() <= i + 1 ||
!strings::safe_strto64(pieces[i + 1], &opts->min_residual_bytes)) {
return ReturnError(pieces, i);
}
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[4]) {
if (pieces.size() <= i + 1 ||
!strings::safe_strto64(pieces[i + 1], &opts->min_output_bytes)) {
return ReturnError(pieces, i);
}
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[5]) {
if (pieces.size() <= i + 1 ||
!strings::safe_strto64(pieces[i + 1], &opts->min_micros)) {
return ReturnError(pieces, i);
}
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[6]) {
if (pieces.size() <= i + 1 ||
!strings::safe_strto64(pieces[i + 1],
&opts->min_accelerator_micros)) {
return ReturnError(pieces, i);
}
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[7]) {
if (pieces.size() <= i + 1 ||
!strings::safe_strto64(pieces[i + 1], &opts->min_cpu_micros)) {
return ReturnError(pieces, i);
}
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[8]) {
if (pieces.size() <= i + 1 ||
!strings::safe_strto64(pieces[i + 1], &opts->min_params)) {
return ReturnError(pieces, i);
}
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[9]) {
if (pieces.size() <= i + 1 ||
!strings::safe_strto64(pieces[i + 1], &opts->min_float_ops)) {
return ReturnError(pieces, i);
}
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[10]) {
if (pieces.size() <= i + 1 ||
!strings::safe_strto64(pieces[i + 1], &opts->min_occurrence)) {
return ReturnError(pieces, i);
}
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[11]) {
if (pieces.size() <= i + 1 ||
!strings::safe_strto64(pieces[i + 1], &opts->step)) {
return ReturnError(pieces, i);
}
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[12]) {
if (pieces.size() <= i + 1) {
return ReturnError(pieces, i);
}
std::set<string> order_by_set(
kOrderBy, kOrderBy + sizeof(kOrderBy) / sizeof(*kOrderBy));
auto order_by = order_by_set.find(pieces[i + 1]);
if (order_by == order_by_set.end()) {
return ReturnError(pieces, i);
}
opts->order_by = *order_by;
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[13]) {
if (pieces.size() <= i + 1) {
return ReturnError(pieces, i);
}
opts->account_type_regexes = str_util::Split(StripQuote(pieces[i + 1]),
',', str_util::SkipEmpty());
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[14]) {
if (pieces.size() <= i + 1) {
return ReturnError(pieces, i);
}
opts->start_name_regexes = str_util::Split(StripQuote(pieces[i + 1]), ',',
str_util::SkipEmpty());
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[15]) {
if (pieces.size() <= i + 1) {
return ReturnError(pieces, i);
}
opts->trim_name_regexes = str_util::Split(StripQuote(pieces[i + 1]), ',',
str_util::SkipEmpty());
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[16]) {
if (pieces.size() <= i + 1) {
return ReturnError(pieces, i);
}
opts->show_name_regexes = str_util::Split(StripQuote(pieces[i + 1]), ',',
str_util::SkipEmpty());
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[17]) {
if (pieces.size() <= i + 1) {
return ReturnError(pieces, i);
}
opts->hide_name_regexes = str_util::Split(StripQuote(pieces[i + 1]), ',',
str_util::SkipEmpty());
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[18]) {
if ((pieces.size() > i + 1 && pieces[i + 1].find("-") == 0) ||
pieces.size() == i + 1) {
opts->account_displayed_op_only = true;
} else if (!StringToBool(pieces[i + 1],
&opts->account_displayed_op_only)) {
return ReturnError(pieces, i);
} else {
++i;
}
} else if (pieces[i] == tensorflow::tfprof::kOptions[19]) {
if (pieces.size() <= i + 1) {
return ReturnError(pieces, i);
}
std::set<string> shown_set(kShown,
kShown + sizeof(kShown) / sizeof(*kShown));
std::vector<string> requested_vector = str_util::Split(
StripQuote(pieces[i + 1]), ',', str_util::SkipEmpty());
std::set<string> requested_set(requested_vector.begin(),
requested_vector.end());
for (const string& requested : requested_set) {
if (shown_set.find(requested) == shown_set.end()) {
return ReturnError(pieces, i);
}
}
opts->select = requested_set;
++i;
} else if (pieces[i] == tensorflow::tfprof::kOptions[20]) {
if (pieces.size() <= i + 1) {
return ReturnError(pieces, i);
}
tensorflow::Status s =
ParseOutput(pieces[i + 1], &opts->output_type, &opts->output_options);
if (!s.ok()) return s;
++i;
} else {
return ReturnError(pieces, i);
}
}
return tensorflow::Status::OK();
}