diff options
Diffstat (limited to 'tensorflow/core/util/command_line_flags.h')
-rw-r--r-- | tensorflow/core/util/command_line_flags.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/core/util/command_line_flags.h b/tensorflow/core/util/command_line_flags.h index 2c77d7874f..f349df16fd 100644 --- a/tensorflow/core/util/command_line_flags.h +++ b/tensorflow/core/util/command_line_flags.h @@ -65,6 +65,7 @@ class Flag { Flag(const char* name, int64* dst1, const string& usage_text); Flag(const char* name, bool* dst, const string& usage_text); Flag(const char* name, string* dst, const string& usage_text); + Flag(const char* name, float* dst, const string& usage_text); private: friend class Flags; @@ -72,11 +73,12 @@ class Flag { bool Parse(string arg, bool* value_parsing_ok) const; string name_; - enum { TYPE_INT, TYPE_INT64, TYPE_BOOL, TYPE_STRING } type_; + enum { TYPE_INT, TYPE_INT64, TYPE_BOOL, TYPE_STRING, TYPE_FLOAT } type_; int* int_value_; int64* int64_value_; bool* bool_value_; string* string_value_; + float* float_value_; string usage_text_; }; |