Registers a flag whose value can be the name of enum members.
tf.compat.v1.flags.DEFINE_enum_class( name, default, enum_class, help, flag_values=_flagvalues.FLAGS, module_name=None, case_sensitive=False, **args )
Args | |
---|---|
name | str, the flag name. |
default | Enum|str|None, the default value of the flag. |
enum_class | class, the Enum class with all the possible values for the flag. |
help | str, the help message. |
flag_values | FlagValues, the FlagValues instance with which the flag will be registered. This should almost never need to be overridden. |
module_name | str, the name of the Python module declaring this flag. If not provided, it will be computed using the stack trace of this call. |
case_sensitive | bool, whether to map strings to members of the enum_class without considering case. |
**args | dict, the extra keyword args that are passed to Flag init. |
Returns | |
---|---|
a handle to defined flag. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/compat/v1/flags/DEFINE_enum_class