1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
/* Copyright 2016 The TensorFlow Authors All Rights Reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
#ifndef THIRD_PARTY_TENSORFLOW_TOOLS_TFPROF_INTERNAL_TFPROF_OPTIONS_H_
#define THIRD_PARTY_TENSORFLOW_TOOLS_TFPROF_INTERNAL_TFPROF_OPTIONS_H_
#include <set>
#include <string>
#include <utility>
#include <vector>
#include "tensorflow/core/framework/types.h"
#include "tensorflow/core/lib/core/status.h"
namespace tensorflow {
namespace tfprof {
static const char* const kOptions[] = {
"-max_depth",
"-min_bytes",
"-min_micros",
"-min_params",
"-min_float_ops",
"-min_occurrence",
"-step",
"-order_by",
"-account_type_regexes",
"-start_name_regexes",
"-trim_name_regexes",
"-show_name_regexes",
"-hide_name_regexes",
"-account_displayed_op_only",
"-select",
"-output",
};
static const char* const kOrderBy[] = {
"name", "bytes", "micros", "accelerator_micros",
"cpu_micros", "params", "float_ops", "occurrence",
};
// Append Only.
// TODO(xpan): As we are adding more fields to be selected, we
// need to have a way to tell users what fields are available in which view.
static const char* const kShown[] = {
"bytes", "micros", "params", "float_ops", "tensor_value",
"device", "op_types", "occurrence", "input_shapes", "accelerator_micros",
"cpu_micros"};
static const char* const kCmds[] = {
"scope", "graph", "code", "op", "set", "help",
};
static const char* const kOutput[] = {"timeline", "stdout", "file"};
static const char* const kTimelineOpts[] = {
"outfile",
};
static const char* const kTimelineRequiredOpts[] = {"outfile"};
static const char* const kFileOpts[] = {
"outfile",
};
static const char* const kFileRequiredOpts[] = {
"outfile",
};
struct Options {
public:
static tensorflow::Status FromProtoStr(const string& opts_proto_str,
Options* opts);
virtual ~Options() {}
Options()
: Options(0, 0, 0, 0, 0, 0, 0, "", {}, {}, {}, {}, {}, false, {}, "",
{}) {}
Options(int max_depth, tensorflow::int64 min_bytes,
tensorflow::int64 min_micros, tensorflow::int64 min_params,
tensorflow::int64 min_float_ops, tensorflow::int64 min_occurrence,
tensorflow::int64 step, const string& order_by,
const std::vector<string>& account_type_regexes,
const std::vector<string>& start_name_regexes,
const std::vector<string>& trim_name_regexes,
const std::vector<string>& show_name_regexes,
const std::vector<string>& hide_name_regexes,
bool account_displayed_op_only, const std::vector<string>& select,
const string& output_type,
const std::map<string, string>& output_options)
: max_depth(max_depth),
min_bytes(min_bytes),
min_micros(min_micros),
min_params(min_params),
min_float_ops(min_float_ops),
min_occurrence(min_occurrence),
step(step),
order_by(order_by),
account_type_regexes(account_type_regexes),
start_name_regexes(start_name_regexes),
trim_name_regexes(trim_name_regexes),
show_name_regexes(show_name_regexes),
hide_name_regexes(hide_name_regexes),
account_displayed_op_only(account_displayed_op_only),
select(select.begin(), select.end()),
output_type(output_type),
output_options(output_options) {}
string ToString() const;
int max_depth;
tensorflow::int64 min_bytes;
tensorflow::int64 min_micros;
tensorflow::int64 min_params;
tensorflow::int64 min_float_ops;
tensorflow::int64 min_occurrence;
tensorflow::int64 step;
string order_by;
std::vector<string> account_type_regexes;
std::vector<string> start_name_regexes;
std::vector<string> trim_name_regexes;
std::vector<string> show_name_regexes;
std::vector<string> hide_name_regexes;
bool account_displayed_op_only;
std::set<string> select;
string output_type;
std::map<string, string> output_options;
};
// Parse the -output option.
// 'output_opt': User input string with format: output_type:key=value,key=value.
// 'output_type' and 'output_options' are extracted from 'output_opt'.
tensorflow::Status ParseOutput(const string& output_opt, string* output_type,
std::map<string, string>* output_options);
} // namespace tfprof
} // namespace tensorflow
#endif // THIRD_PARTY_TENSORFLOW_TOOLS_TFPROF_INTERNAL_TFPROF_OPTIONS_H_
|
