tf.compat.v1.flags.DEFINE_enum_class

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,
    required=False,
    **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 :class: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.
required bool, is this a required flag. This must be used as a keyword argument.
**args dict, the extra keyword args that are passed to Flag.__init__.

a handle to defined flag.