tensorbored 2.21.0rc1769983804__py3-none-any.whl

tensorbored/compat/proto/types_pb2.py

@@ -0,0 +1,105 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/types.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n$tensorbored/compat/proto/types.proto\x12\x0btensorbored\":\n\x0fSerializedDType\x12\'\n\x08\x64\x61tatype\x18\x01 \x01(\x0e\x32\x15.tensorbored.DataType*\xa9\t\n\x08\x44\x61taType\x12\x0e\n\nDT_INVALID\x10\x00\x12\x0c\n\x08\x44T_FLOAT\x10\x01\x12\r\n\tDT_DOUBLE\x10\x02\x12\x0c\n\x08\x44T_INT32\x10\x03\x12\x0c\n\x08\x44T_UINT8\x10\x04\x12\x0c\n\x08\x44T_INT16\x10\x05\x12\x0b\n\x07\x44T_INT8\x10\x06\x12\r\n\tDT_STRING\x10\x07\x12\x10\n\x0c\x44T_COMPLEX64\x10\x08\x12\x0c\n\x08\x44T_INT64\x10\t\x12\x0b\n\x07\x44T_BOOL\x10\n\x12\x0c\n\x08\x44T_QINT8\x10\x0b\x12\r\n\tDT_QUINT8\x10\x0c\x12\r\n\tDT_QINT32\x10\r\x12\x0f\n\x0b\x44T_BFLOAT16\x10\x0e\x12\r\n\tDT_QINT16\x10\x0f\x12\x0e\n\nDT_QUINT16\x10\x10\x12\r\n\tDT_UINT16\x10\x11\x12\x11\n\rDT_COMPLEX128\x10\x12\x12\x0b\n\x07\x44T_HALF\x10\x13\x12\x0f\n\x0b\x44T_RESOURCE\x10\x14\x12\x0e\n\nDT_VARIANT\x10\x15\x12\r\n\tDT_UINT32\x10\x16\x12\r\n\tDT_UINT64\x10\x17\x12\x12\n\x0e\x44T_FLOAT8_E5M2\x10\x18\x12\x14\n\x10\x44T_FLOAT8_E4M3FN\x10\x19\x12\x16\n\x12\x44T_FLOAT8_E4M3FNUZ\x10\x1a\x12\x19\n\x15\x44T_FLOAT8_E4M3B11FNUZ\x10\x1b\x12\x16\n\x12\x44T_FLOAT8_E5M2FNUZ\x10\x1c\x12\x0b\n\x07\x44T_INT4\x10\x1d\x12\x0c\n\x08\x44T_UINT4\x10\x1e\x12\x0b\n\x07\x44T_INT2\x10\x1f\x12\x0c\n\x08\x44T_UINT2\x10 \x12\x10\n\x0c\x44T_FLOAT_REF\x10\x65\x12\x11\n\rDT_DOUBLE_REF\x10\x66\x12\x10\n\x0c\x44T_INT32_REF\x10g\x12\x10\n\x0c\x44T_UINT8_REF\x10h\x12\x10\n\x0c\x44T_INT16_REF\x10i\x12\x0f\n\x0b\x44T_INT8_REF\x10j\x12\x11\n\rDT_STRING_REF\x10k\x12\x14\n\x10\x44T_COMPLEX64_REF\x10l\x12\x10\n\x0c\x44T_INT64_REF\x10m\x12\x0f\n\x0b\x44T_BOOL_REF\x10n\x12\x10\n\x0c\x44T_QINT8_REF\x10o\x12\x11\n\rDT_QUINT8_REF\x10p\x12\x11\n\rDT_QINT32_REF\x10q\x12\x13\n\x0f\x44T_BFLOAT16_REF\x10r\x12\x11\n\rDT_QINT16_REF\x10s\x12\x12\n\x0e\x44T_QUINT16_REF\x10t\x12\x11\n\rDT_UINT16_REF\x10u\x12\x15\n\x11\x44T_COMPLEX128_REF\x10v\x12\x0f\n\x0b\x44T_HALF_REF\x10w\x12\x13\n\x0f\x44T_RESOURCE_REF\x10x\x12\x12\n\x0e\x44T_VARIANT_REF\x10y\x12\x11\n\rDT_UINT32_REF\x10z\x12\x11\n\rDT_UINT64_REF\x10{\x12\x16\n\x12\x44T_FLOAT8_E5M2_REF\x10|\x12\x18\n\x14\x44T_FLOAT8_E4M3FN_REF\x10}\x12\x1a\n\x16\x44T_FLOAT8_E4M3FNUZ_REF\x10~\x12\x1d\n\x19\x44T_FLOAT8_E4M3B11FNUZ_REF\x10\x7f\x12\x1b\n\x16\x44T_FLOAT8_E5M2FNUZ_REF\x10\x80\x01\x12\x10\n\x0b\x44T_INT4_REF\x10\x81\x01\x12\x11\n\x0c\x44T_UINT4_REF\x10\x82\x01\x12\x10\n\x0b\x44T_INT2_REF\x10\x83\x01\x12\x11\n\x0c\x44T_UINT2_REF\x10\x84\x01\x42z\n\x18org.tensorflow.frameworkB\x0bTypesProtosP\x01ZLgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/types_go_proto\xf8\x01\x01\x62\x06proto3')
+
+_DATATYPE = DESCRIPTOR.enum_types_by_name['DataType']
+DataType = enum_type_wrapper.EnumTypeWrapper(_DATATYPE)
+DT_INVALID = 0
+DT_FLOAT = 1
+DT_DOUBLE = 2
+DT_INT32 = 3
+DT_UINT8 = 4
+DT_INT16 = 5
+DT_INT8 = 6
+DT_STRING = 7
+DT_COMPLEX64 = 8
+DT_INT64 = 9
+DT_BOOL = 10
+DT_QINT8 = 11
+DT_QUINT8 = 12
+DT_QINT32 = 13
+DT_BFLOAT16 = 14
+DT_QINT16 = 15
+DT_QUINT16 = 16
+DT_UINT16 = 17
+DT_COMPLEX128 = 18
+DT_HALF = 19
+DT_RESOURCE = 20
+DT_VARIANT = 21
+DT_UINT32 = 22
+DT_UINT64 = 23
+DT_FLOAT8_E5M2 = 24
+DT_FLOAT8_E4M3FN = 25
+DT_FLOAT8_E4M3FNUZ = 26
+DT_FLOAT8_E4M3B11FNUZ = 27
+DT_FLOAT8_E5M2FNUZ = 28
+DT_INT4 = 29
+DT_UINT4 = 30
+DT_INT2 = 31
+DT_UINT2 = 32
+DT_FLOAT_REF = 101
+DT_DOUBLE_REF = 102
+DT_INT32_REF = 103
+DT_UINT8_REF = 104
+DT_INT16_REF = 105
+DT_INT8_REF = 106
+DT_STRING_REF = 107
+DT_COMPLEX64_REF = 108
+DT_INT64_REF = 109
+DT_BOOL_REF = 110
+DT_QINT8_REF = 111
+DT_QUINT8_REF = 112
+DT_QINT32_REF = 113
+DT_BFLOAT16_REF = 114
+DT_QINT16_REF = 115
+DT_QUINT16_REF = 116
+DT_UINT16_REF = 117
+DT_COMPLEX128_REF = 118
+DT_HALF_REF = 119
+DT_RESOURCE_REF = 120
+DT_VARIANT_REF = 121
+DT_UINT32_REF = 122
+DT_UINT64_REF = 123
+DT_FLOAT8_E5M2_REF = 124
+DT_FLOAT8_E4M3FN_REF = 125
+DT_FLOAT8_E4M3FNUZ_REF = 126
+DT_FLOAT8_E4M3B11FNUZ_REF = 127
+DT_FLOAT8_E5M2FNUZ_REF = 128
+DT_INT4_REF = 129
+DT_UINT4_REF = 130
+DT_INT2_REF = 131
+DT_UINT2_REF = 132
+
+
+_SERIALIZEDDTYPE = DESCRIPTOR.message_types_by_name['SerializedDType']
+SerializedDType = _reflection.GeneratedProtocolMessageType('SerializedDType', (_message.Message,), {
+  'DESCRIPTOR' : _SERIALIZEDDTYPE,
+  '__module__' : 'tensorbored.compat.proto.types_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SerializedDType)
+  })
+_sym_db.RegisterMessage(SerializedDType)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\013TypesProtosP\001ZLgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/types_go_proto\370\001\001'
+  _DATATYPE._serialized_start=114
+  _DATATYPE._serialized_end=1307
+  _SERIALIZEDDTYPE._serialized_start=53
+  _SERIALIZEDDTYPE._serialized_end=111
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/variable_pb2.py

@@ -0,0 +1,62 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/variable.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\'tensorbored/compat/proto/variable.proto\x12\x0btensorbored\"\xcb\x02\n\x0bVariableDef\x12\x15\n\rvariable_name\x18\x01 \x01(\t\x12\x1a\n\x12initial_value_name\x18\x06 \x01(\t\x12\x18\n\x10initializer_name\x18\x02 \x01(\t\x12\x15\n\rsnapshot_name\x18\x03 \x01(\t\x12:\n\x13save_slice_info_def\x18\x04 \x01(\x0b\x32\x1d.tensorbored.SaveSliceInfoDef\x12\x13\n\x0bis_resource\x18\x05 \x01(\x08\x12\x11\n\ttrainable\x18\x07 \x01(\x08\x12=\n\x0fsynchronization\x18\x08 \x01(\x0e\x32$.tensorbored.VariableSynchronization\x12\x35\n\x0b\x61ggregation\x18\t \x01(\x0e\x32 .tensorbored.VariableAggregation\"`\n\x10SaveSliceInfoDef\x12\x11\n\tfull_name\x18\x01 \x01(\t\x12\x12\n\nfull_shape\x18\x02 \x03(\x03\x12\x12\n\nvar_offset\x18\x03 \x03(\x03\x12\x11\n\tvar_shape\x18\x04 \x03(\x03*\xac\x01\n\x17VariableSynchronization\x12!\n\x1dVARIABLE_SYNCHRONIZATION_AUTO\x10\x00\x12!\n\x1dVARIABLE_SYNCHRONIZATION_NONE\x10\x01\x12%\n!VARIABLE_SYNCHRONIZATION_ON_WRITE\x10\x02\x12$\n VARIABLE_SYNCHRONIZATION_ON_READ\x10\x03*\x9e\x01\n\x13VariableAggregation\x12\x1d\n\x19VARIABLE_AGGREGATION_NONE\x10\x00\x12\x1c\n\x18VARIABLE_AGGREGATION_SUM\x10\x01\x12\x1d\n\x19VARIABLE_AGGREGATION_MEAN\x10\x02\x12+\n\'VARIABLE_AGGREGATION_ONLY_FIRST_REPLICA\x10\x03\x42\x80\x01\n\x18org.tensorflow.frameworkB\x0eVariableProtosP\x01ZOgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/variable_go_proto\xf8\x01\x01\x62\x06proto3')
+
+_VARIABLESYNCHRONIZATION = DESCRIPTOR.enum_types_by_name['VariableSynchronization']
+VariableSynchronization = enum_type_wrapper.EnumTypeWrapper(_VARIABLESYNCHRONIZATION)
+_VARIABLEAGGREGATION = DESCRIPTOR.enum_types_by_name['VariableAggregation']
+VariableAggregation = enum_type_wrapper.EnumTypeWrapper(_VARIABLEAGGREGATION)
+VARIABLE_SYNCHRONIZATION_AUTO = 0
+VARIABLE_SYNCHRONIZATION_NONE = 1
+VARIABLE_SYNCHRONIZATION_ON_WRITE = 2
+VARIABLE_SYNCHRONIZATION_ON_READ = 3
+VARIABLE_AGGREGATION_NONE = 0
+VARIABLE_AGGREGATION_SUM = 1
+VARIABLE_AGGREGATION_MEAN = 2
+VARIABLE_AGGREGATION_ONLY_FIRST_REPLICA = 3
+
+
+_VARIABLEDEF = DESCRIPTOR.message_types_by_name['VariableDef']
+_SAVESLICEINFODEF = DESCRIPTOR.message_types_by_name['SaveSliceInfoDef']
+VariableDef = _reflection.GeneratedProtocolMessageType('VariableDef', (_message.Message,), {
+  'DESCRIPTOR' : _VARIABLEDEF,
+  '__module__' : 'tensorbored.compat.proto.variable_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.VariableDef)
+  })
+_sym_db.RegisterMessage(VariableDef)
+
+SaveSliceInfoDef = _reflection.GeneratedProtocolMessageType('SaveSliceInfoDef', (_message.Message,), {
+  'DESCRIPTOR' : _SAVESLICEINFODEF,
+  '__module__' : 'tensorbored.compat.proto.variable_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.SaveSliceInfoDef)
+  })
+_sym_db.RegisterMessage(SaveSliceInfoDef)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\016VariableProtosP\001ZOgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/variable_go_proto\370\001\001'
+  _VARIABLESYNCHRONIZATION._serialized_start=489
+  _VARIABLESYNCHRONIZATION._serialized_end=661
+  _VARIABLEAGGREGATION._serialized_start=664
+  _VARIABLEAGGREGATION._serialized_end=822
+  _VARIABLEDEF._serialized_start=57
+  _VARIABLEDEF._serialized_end=388
+  _SAVESLICEINFODEF._serialized_start=390
+  _SAVESLICEINFODEF._serialized_end=486
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/verifier_config_pb2.py

@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/verifier_config.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n.tensorbored/compat/proto/verifier_config.proto\x12\x0btensorbored\"\x9c\x01\n\x0eVerifierConfig\x12\"\n\x1averification_timeout_in_ms\x18\x01 \x01(\x03\x12>\n\x12structure_verifier\x18\x02 \x01(\x0e\x32\".tensorbored.VerifierConfig.Toggle\"&\n\x06Toggle\x12\x0b\n\x07\x44\x45\x46\x41ULT\x10\x00\x12\x06\n\x02ON\x10\x01\x12\x07\n\x03OFF\x10\x02\x42\x8c\x01\n\x18org.tensorflow.frameworkB\x14VerifierConfigProtosP\x01ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_VERIFIERCONFIG = DESCRIPTOR.message_types_by_name['VerifierConfig']
+_VERIFIERCONFIG_TOGGLE = _VERIFIERCONFIG.enum_types_by_name['Toggle']
+VerifierConfig = _reflection.GeneratedProtocolMessageType('VerifierConfig', (_message.Message,), {
+  'DESCRIPTOR' : _VERIFIERCONFIG,
+  '__module__' : 'tensorbored.compat.proto.verifier_config_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.VerifierConfig)
+  })
+_sym_db.RegisterMessage(VerifierConfig)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\024VerifierConfigProtosP\001ZUgithub.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto\370\001\001'
+  _VERIFIERCONFIG._serialized_start=64
+  _VERIFIERCONFIG._serialized_end=220
+  _VERIFIERCONFIG_TOGGLE._serialized_start=182
+  _VERIFIERCONFIG_TOGGLE._serialized_end=220
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/versions_pb2.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/versions.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\'tensorbored/compat/proto/versions.proto\x12\x0btensorbored\"K\n\nVersionDef\x12\x10\n\x08producer\x18\x01 \x01(\x05\x12\x14\n\x0cmin_consumer\x18\x02 \x01(\x05\x12\x15\n\rbad_consumers\x18\x03 \x03(\x05\x42\x80\x01\n\x18org.tensorflow.frameworkB\x0eVersionsProtosP\x01ZOgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/versions_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_VERSIONDEF = DESCRIPTOR.message_types_by_name['VersionDef']
+VersionDef = _reflection.GeneratedProtocolMessageType('VersionDef', (_message.Message,), {
+  'DESCRIPTOR' : _VERSIONDEF,
+  '__module__' : 'tensorbored.compat.proto.versions_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.VersionDef)
+  })
+_sym_db.RegisterMessage(VersionDef)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\016VersionsProtosP\001ZOgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/versions_go_proto\370\001\001'
+  _VERSIONDEF._serialized_start=56
+  _VERSIONDEF._serialized_end=131
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/tensorflow_stub/__init__.py

@@ -0,0 +1,38 @@
+# Copyright 2018 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.
+# ==============================================================================
+
+
+from tensorbored.compat.proto.config_pb2 import *  # noqa
+from tensorbored.compat.proto.event_pb2 import *  # noqa
+from tensorbored.compat.proto.graph_pb2 import *  # noqa
+from tensorbored.compat.proto.meta_graph_pb2 import *  # noqa
+from tensorbored.compat.proto.summary_pb2 import *  # noqa
+from .dtypes import as_dtype  # noqa
+from .dtypes import DType  # noqa
+from .dtypes import string  # noqa
+from . import app  # noqa
+from . import compat  # noqa
+from . import dtypes  # noqa
+from . import error_codes  # noqa
+from . import errors  # noqa
+from . import flags  # noqa
+from . import io  # noqa
+from . import pywrap_tensorflow  # noqa
+from . import tensor_shape  # noqa
+
+compat.v1.errors = errors
+
+# Set a fake __version__ to help distinguish this as our own stub API.
+__version__ = "stub"

tensorbored/compat/tensorflow_stub/app.py

@@ -0,0 +1,124 @@
+# Copyright 2015 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.
+# ==============================================================================
+
+"""Generic entry point script."""
+
+import errno as _errno
+import sys as _sys
+
+from . import flags
+
+
+def _usage(shorthelp):
+    """Writes __main__'s docstring to stdout with some help text.
+
+    Args:
+      shorthelp: bool, if True, prints only flags from the main module,
+          rather than all flags.
+    """
+    doc = _sys.modules["__main__"].__doc__
+    if not doc:
+        doc = "\nUSAGE: %s [flags]\n" % _sys.argv[0]
+        doc = flags.text_wrap(doc, indent="       ", firstline_indent="")
+    else:
+        # Replace all '%s' with sys.argv[0], and all '%%' with '%'.
+        num_specifiers = doc.count("%") - 2 * doc.count("%%")
+        try:
+            doc %= (_sys.argv[0],) * num_specifiers
+        except (OverflowError, TypeError, ValueError):
+            # Just display the docstring as-is.
+            pass
+    if shorthelp:
+        flag_str = flags.FLAGS.main_module_help()
+    else:
+        flag_str = str(flags.FLAGS)
+    try:
+        _sys.stdout.write(doc)
+        if flag_str:
+            _sys.stdout.write("\nflags:\n")
+            _sys.stdout.write(flag_str)
+        _sys.stdout.write("\n")
+    except IOError as e:
+        # We avoid printing a huge backtrace if we get EPIPE, because
+        # "foo.par --help | less" is a frequent use case.
+        if e.errno != _errno.EPIPE:
+            raise
+
+
+class _HelpFlag(flags.BooleanFlag):
+    """Special boolean flag that displays usage and raises SystemExit."""
+
+    NAME = "help"
+    SHORT_NAME = "h"
+
+    def __init__(self):
+        super().__init__(
+            self.NAME, False, "show this help", short_name=self.SHORT_NAME
+        )
+
+    def parse(self, arg):
+        if arg:
+            _usage(shorthelp=True)
+            print()
+            print("Try --helpfull to get a list of all flags.")
+            _sys.exit(1)
+
+
+class _HelpshortFlag(_HelpFlag):
+    """--helpshort is an alias for --help."""
+
+    NAME = "helpshort"
+    SHORT_NAME = None
+
+
+class _HelpfullFlag(flags.BooleanFlag):
+    """Display help for flags in main module and all dependent modules."""
+
+    def __init__(self):
+        super().__init__("helpfull", False, "show full help")
+
+    def parse(self, arg):
+        if arg:
+            _usage(shorthelp=False)
+            _sys.exit(1)
+
+
+_define_help_flags_called = False
+
+
+def _define_help_flags():
+    global _define_help_flags_called
+    if not _define_help_flags_called:
+        flags.DEFINE_flag(_HelpFlag())
+        flags.DEFINE_flag(_HelpfullFlag())
+        flags.DEFINE_flag(_HelpshortFlag())
+        _define_help_flags_called = True
+
+
+# @tf_export('app.run')
+def run(main=None, argv=None):
+    """Runs the program with an optional 'main' function and 'argv' list."""
+
+    # Define help flags.
+    _define_help_flags()
+
+    # Parse known flags.
+    argv = flags.FLAGS(_sys.argv if argv is None else argv, known_only=True)
+
+    main = main or _sys.modules["__main__"].main
+
+    # Call the main function, passing through any arguments
+    # to the final program.
+    _sys.exit(main(argv))

tensorbored/compat/tensorflow_stub/compat/__init__.py

@@ -0,0 +1,131 @@
+# Copyright 2015 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.
+# ==============================================================================
+"""Functions for Python 2 vs. 3 compatibility.
+
+## Conversion routines
+In addition to the functions below, `as_str` converts an object to a `str`.
+
+
+## Types
+The compatibility module also provides the following types:
+
+* `bytes_or_text_types`
+* `complex_types`
+* `integral_types`
+* `real_types`
+"""
+
+import numbers as _numbers
+import numpy as _np
+
+from tensorbored.compat.tensorflow_stub.compat.v1 import *  # noqa
+
+
+def as_bytes(bytes_or_text, encoding="utf-8"):
+    """Converts either bytes or unicode to `bytes`, using utf-8 encoding for
+    text.
+
+    Args:
+    bytes_or_text: A `bytes`, `str`, or `unicode` object.
+    encoding: A string indicating the charset for encoding unicode.
+
+    Returns:
+    A `bytes` object.
+
+    Raises:
+    TypeError: If `bytes_or_text` is not a binary or unicode string.
+    """
+    if isinstance(bytes_or_text, str):
+        return bytes_or_text.encode(encoding)
+    elif isinstance(bytes_or_text, bytes):
+        return bytes_or_text
+    else:
+        raise TypeError(
+            "Expected binary or unicode string, got %r" % (bytes_or_text,)
+        )
+
+
+def as_text(bytes_or_text, encoding="utf-8"):
+    """Returns the given argument as a unicode string.
+
+    Args:
+    bytes_or_text: A `bytes`, `str`, or `unicode` object.
+    encoding: A string indicating the charset for decoding unicode.
+
+    Returns:
+    A `unicode` (Python 2) or `str` (Python 3) object.
+
+    Raises:
+    TypeError: If `bytes_or_text` is not a binary or unicode string.
+    """
+    if isinstance(bytes_or_text, str):
+        return bytes_or_text
+    elif isinstance(bytes_or_text, bytes):
+        return bytes_or_text.decode(encoding)
+    else:
+        raise TypeError(
+            "Expected binary or unicode string, got %r" % bytes_or_text
+        )
+
+
+# Convert an object to a `str` in both Python 2 and 3.
+as_str = as_text
+
+
+# @tf_export('compat.as_str_any')
+def as_str_any(value):
+    """Converts to `str` as `str(value)`, but use `as_str` for `bytes`.
+
+    Args:
+    value: A object that can be converted to `str`.
+
+    Returns:
+    A `str` object.
+    """
+    if isinstance(value, bytes):
+        return as_str(value)
+    else:
+        return str(value)
+
+
+# @tf_export('compat.path_to_str')
+def path_to_str(path):
+    """Returns the file system path representation of a `PathLike` object, else
+    as it is.
+
+    Args:
+    path: An object that can be converted to path representation.
+
+    Returns:
+    A `str` object.
+    """
+    if hasattr(path, "__fspath__"):
+        path = as_str_any(path.__fspath__())
+    return path
+
+
+# Numpy 1.8 scalars don't inherit from numbers.Integral in Python 3, so we
+# need to check them specifically.  The same goes from Real and Complex.
+integral_types = (_numbers.Integral, _np.integer)
+# tf_export('compat.integral_types').export_constant(__name__, 'integral_types')
+real_types = (_numbers.Real, _np.integer, _np.floating)
+# tf_export('compat.real_types').export_constant(__name__, 'real_types')
+complex_types = (_numbers.Complex, _np.number)
+# tf_export('compat.complex_types').export_constant(__name__, 'complex_types')
+
+# Either bytes or text.
+bytes_or_text_types = (bytes, str)
+# tf_export('compat.bytes_or_text_types').export_constant(__name__,
+#                                                         'bytes_or_text_types')

tensorbored/compat/tensorflow_stub/compat/v1/__init__.py

@@ -0,0 +1,20 @@
+# Copyright 2018 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.
+# ==============================================================================
+
+
+# Set this in tensorboard/compat/tensorflow_stub/__init__.py to eliminate
+# any cycles on import
+#
+# from tensorbored.compat.tensorflow_stub import pywrap_tensorflow  # noqa