tensorbored 2.21.0rc1769983804__py3-none-any.whl

tensorbored/backend/json_util.py

@@ -0,0 +1,70 @@
+# 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.
+# ==============================================================================
+
+"""A module providing a function for serializing JSON values with Infinity.
+
+Python provides no way to override how json.dumps serializes
+Infinity/-Infinity/NaN; if allow_nan is true, it encodes them as
+Infinity/-Infinity/NaN, in violation of the JSON spec and in violation
+of what JSON.parse accepts. If it's false, it throws a ValueError,
+Neither subclassing JSONEncoder nor passing a function in the |default|
+keyword argument overrides this.
+"""
+
+import collections
+import math
+
+_INFINITY = float("inf")
+_NEGATIVE_INFINITY = float("-inf")
+
+
+def Cleanse(obj, encoding="utf-8"):
+    """Makes Python object appropriate for JSON serialization.
+
+    - Replaces instances of Infinity/-Infinity/NaN with strings.
+    - Turns byte strings into unicode strings.
+    - Turns sets into sorted lists.
+    - Turns tuples into lists.
+
+    Args:
+      obj: Python data structure.
+      encoding: Charset used to decode byte strings.
+
+    Returns:
+      Unicode JSON data structure.
+    """
+    if isinstance(obj, int):
+        return obj
+    elif isinstance(obj, float):
+        if obj == _INFINITY:
+            return "Infinity"
+        elif obj == _NEGATIVE_INFINITY:
+            return "-Infinity"
+        elif math.isnan(obj):
+            return "NaN"
+        else:
+            return obj
+    elif isinstance(obj, bytes):
+        return obj.decode(encoding)
+    elif isinstance(obj, (list, tuple)):
+        return [Cleanse(i, encoding) for i in obj]
+    elif isinstance(obj, set):
+        return [Cleanse(i, encoding) for i in sorted(obj)]
+    elif isinstance(obj, dict):
+        return collections.OrderedDict(
+            (Cleanse(k, encoding), Cleanse(v, encoding)) for k, v in obj.items()
+        )
+    else:
+        return obj

tensorbored/backend/path_prefix.py

@@ -0,0 +1,67 @@
+# Copyright 2019 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.
+# ==============================================================================
+"""Internal path prefix support for TensorBoard.
+
+Using a path prefix of `/foo/bar` enables TensorBoard to serve from
+`http://localhost:6006/foo/bar/` rather than `http://localhost:6006/`.
+See the `--path_prefix` flag docs for more details.
+"""
+
+from tensorbored import errors
+
+
+class PathPrefixMiddleware:
+    """WSGI middleware for path prefixes.
+
+    All requests to this middleware must begin with the specified path
+    prefix (otherwise, a 404 will be returned immediately). Requests
+    will be forwarded to the underlying application with the path prefix
+    stripped and appended to `SCRIPT_NAME` (see the WSGI spec, PEP 3333,
+    for details).
+    """
+
+    def __init__(self, application, path_prefix):
+        """Initializes this middleware.
+
+        Args:
+          application: The WSGI application to wrap (see PEP 3333).
+          path_prefix: A string path prefix to be stripped from incoming
+            requests. If empty, this middleware is a no-op. If non-empty,
+            the path prefix must start with a slash and not end with one
+            (e.g., "/tensorboard").
+        """
+        if path_prefix.endswith("/"):
+            raise ValueError(
+                "Path prefix must not end with slash: %r" % path_prefix
+            )
+        if path_prefix and not path_prefix.startswith("/"):
+            raise ValueError(
+                "Non-empty path prefix must start with slash: %r" % path_prefix
+            )
+        self._application = application
+        self._path_prefix = path_prefix
+        self._strict_prefix = self._path_prefix + "/"
+
+    def __call__(self, environ, start_response):
+        path = environ.get("PATH_INFO", "")
+        if path != self._path_prefix and not path.startswith(
+            self._strict_prefix
+        ):
+            raise errors.NotFoundError()
+        environ["PATH_INFO"] = path[len(self._path_prefix) :]
+        environ["SCRIPT_NAME"] = (
+            environ.get("SCRIPT_NAME", "") + self._path_prefix
+        )
+        return self._application(environ, start_response)

tensorbored/backend/process_graph.py

@@ -0,0 +1,74 @@
+# 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.
+# ==============================================================================
+"""Graph post-processing logic.
+
+Used by both TensorBoard and mldash.
+"""
+
+
+def prepare_graph_for_ui(
+    graph, limit_attr_size=1024, large_attrs_key="_too_large_attrs"
+):
+    """Prepares (modifies in-place) the graph to be served to the front-end.
+
+    For now, it supports filtering out attributes that are
+    too large to be shown in the graph UI.
+
+    Args:
+      graph: The GraphDef proto message.
+      limit_attr_size: Maximum allowed size in bytes, before the attribute
+          is considered large. Default is 1024 (1KB). Must be > 0 or None.
+          If None, there will be no filtering.
+      large_attrs_key: The attribute key that will be used for storing attributes
+          that are too large. Default is '_too_large_attrs'. Must be != None if
+          `limit_attr_size` is != None.
+
+    Raises:
+      ValueError: If `large_attrs_key is None` while `limit_attr_size != None`.
+      ValueError: If `limit_attr_size` is defined, but <= 0.
+    """
+    # TODO(@davidsoergel): detect whether a graph has been filtered already
+    # (to a limit_attr_size <= what is requested here).  If it is already
+    # filtered, return immediately.
+
+    # Check input for validity.
+    if limit_attr_size is not None:
+        if large_attrs_key is None:
+            raise ValueError(
+                "large_attrs_key must be != None when limit_attr_size"
+                "!= None."
+            )
+
+        if limit_attr_size <= 0:
+            raise ValueError(
+                "limit_attr_size must be > 0, but is %d" % limit_attr_size
+            )
+
+    # Filter only if a limit size is defined.
+    if limit_attr_size is not None:
+        for node in graph.node:
+            # Go through all the attributes and filter out ones bigger than the
+            # limit.
+            keys = list(node.attr.keys())
+            for key in keys:
+                size = node.attr[key].ByteSize()
+                if size > limit_attr_size or size < 0:
+                    del node.attr[key]
+                    # Add the attribute key to the list of "too large" attributes.
+                    # This is used in the info card in the graph UI to show the user
+                    # that some attributes are too large to be shown.
+                    node.attr[large_attrs_key].list.s.append(
+                        key.encode("utf-8")
+                    )

tensorbored/backend/security_validator.py

@@ -0,0 +1,202 @@
+# Copyright 2019 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.
+# ==============================================================================
+"""Validates responses and their security features."""
+
+import dataclasses
+
+from typing import Collection
+
+from werkzeug.datastructures import Headers
+from werkzeug import http
+
+from tensorbored.util import tb_logging
+
+logger = tb_logging.get_logger()
+
+_HTML_MIME_TYPE = "text/html"
+_CSP_DEFAULT_SRC = "default-src"
+# Whitelist of allowed CSP violations.
+_CSP_IGNORE = {
+    # Polymer-based code uses unsafe-inline.
+    "style-src": ["'unsafe-inline'", "data:"],
+    # Used in canvas
+    "img-src": ["blob:", "data:"],
+    # Used by numericjs.
+    # TODO(stephanwlee): remove it eventually.
+    "script-src": ["'unsafe-eval'"],
+    "font-src": ["data:"],
+}
+
+
+@dataclasses.dataclass(frozen=True)
+class Directive:
+    """Content security policy directive.
+
+    Loosely follow vocabulary from https://www.w3.org/TR/CSP/#framework-directives.
+
+    Attributes:
+      name: A non-empty string.
+      value: A collection of non-empty strings.
+    """
+
+    name: str
+    value: Collection[str]
+
+
+def _maybe_raise_value_error(error_msg):
+    logger.warning("In 3.0, this warning will become an error:\n%s" % error_msg)
+    # TODO(3.x): raise a value error.
+
+
+class SecurityValidatorMiddleware:
+    """WSGI middleware validating security on response.
+
+    It validates:
+    - responses have Content-Type
+    - responses have X-Content-Type-Options: nosniff
+    - text/html responses have CSP header. It also validates whether the CSP
+      headers pass basic requirement. e.g., default-src should be present, cannot
+      use "*" directive, and others. For more complete list, please refer to
+      _validate_csp_policies.
+
+    Instances of this class are WSGI applications (see PEP 3333).
+    """
+
+    def __init__(self, application):
+        """Initializes an `SecurityValidatorMiddleware`.
+
+        Args:
+          application: The WSGI application to wrap (see PEP 3333).
+        """
+        self._application = application
+
+    def __call__(self, environ, start_response):
+        def start_response_proxy(status, headers, exc_info=None):
+            self._validate_headers(headers)
+            return start_response(status, headers, exc_info)
+
+        return self._application(environ, start_response_proxy)
+
+    def _validate_headers(self, headers_list):
+        headers = Headers(headers_list)
+        self._validate_content_type(headers)
+        self._validate_x_content_type_options(headers)
+        self._validate_csp_headers(headers)
+
+    def _validate_content_type(self, headers):
+        if headers.get("Content-Type"):
+            return
+
+        _maybe_raise_value_error("Content-Type is required on a Response")
+
+    def _validate_x_content_type_options(self, headers):
+        option = headers.get("X-Content-Type-Options")
+        if option == "nosniff":
+            return
+
+        _maybe_raise_value_error(
+            'X-Content-Type-Options is required to be "nosniff"'
+        )
+
+    def _validate_csp_headers(self, headers):
+        mime_type, _ = http.parse_options_header(headers.get("Content-Type"))
+        if mime_type != _HTML_MIME_TYPE:
+            return
+
+        csp_texts = headers.get_all("Content-Security-Policy")
+        policies = []
+
+        for csp_text in csp_texts:
+            policies += self._parse_serialized_csp(csp_text)
+
+        self._validate_csp_policies(policies)
+
+    def _validate_csp_policies(self, policies):
+        has_default_src = False
+        violations = []
+
+        for directive in policies:
+            name = directive.name
+            for value in directive.value:
+                has_default_src = has_default_src or name == _CSP_DEFAULT_SRC
+
+                if value in _CSP_IGNORE.get(name, []):
+                    # There are cases where certain directives are legitimate.
+                    continue
+
+                # TensorBoard follows principle of least privilege. However, to make it
+                # easier to conform to the security policy for plugin authors,
+                # TensorBoard trusts request and resources originating its server. Also,
+                # it can selectively trust domains as long as they use https protocol.
+                # Lastly, it can allow 'none' directive.
+                # TODO(stephanwlee): allow configuration for whitelist of domains for
+                # stricter enforcement.
+                # TODO(stephanwlee): deprecate the sha-based whitelisting.
+                if (
+                    value == "'self'"
+                    or value == "'none'"
+                    or value.startswith("https:")
+                    or value.startswith("'sha256-")
+                ):
+                    continue
+
+                msg = "Illegal Content-Security-Policy for {name}: {value}".format(
+                    name=name, value=value
+                )
+                violations.append(msg)
+
+        if not has_default_src:
+            violations.append(
+                "Requires default-src for Content-Security-Policy"
+            )
+
+        if violations:
+            _maybe_raise_value_error("\n".join(violations))
+
+    def _parse_serialized_csp(self, csp_text):
+        # See https://www.w3.org/TR/CSP/#parse-serialized-policy.
+        # Below Steps are based on the algorithm stated in above spec.
+        # Deviations:
+        # - it does not warn and ignore duplicative directive (Step 2.5)
+
+        # Step 2
+        csp_srcs = csp_text.split(";")
+
+        policy = []
+        for token in csp_srcs:
+            # Step 2.1
+            token = token.strip()
+
+            if not token:
+                # Step 2.2
+                continue
+
+            # Step 2.3
+            token_frag = token.split(None, 1)
+            name = token_frag[0]
+
+            values = token_frag[1] if len(token_frag) == 2 else ""
+
+            # Step 2.4
+            name = name.lower()
+
+            # Step 2.6
+            value = values.split()
+            # Step 2.7
+            directive = Directive(name=name, value=value)
+            # Step 2.8
+            policy.append(directive)
+
+        return policy

tensorbored/compat/__init__.py

@@ -0,0 +1,69 @@
+# Copyright 2017 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.
+
+"""Compatibility interfaces for TensorBoard.
+
+This module provides logic for importing variations on the TensorFlow
+APIs, as lazily loaded imports to help avoid circular dependency issues
+and defer the search and loading of the module until necessary.
+"""
+
+import tensorbored.lazy as _lazy
+
+
+@_lazy.lazy_load("tensorbored.compat.tf")
+def tf():
+    """Provide the root module of a TF-like API for use within TensorBoard.
+
+    By default this is equivalent to `import tensorflow as tf`, but it can be used
+    in combination with //tensorbored/compat:tensorflow (to fall back to a stub TF
+    API implementation if the real one is not available) or with
+    //tensorbored/compat:no_tensorflow (to force unconditional use of the stub).
+
+    Returns:
+      The root module of a TF-like API, if available.
+
+    Raises:
+      ImportError: if a TF-like API is not available.
+    """
+    try:
+        from tensorbored.compat import notf  # noqa: F401
+    except ImportError:
+        try:
+            import tensorflow
+
+            return tensorflow
+        except ImportError:
+            pass
+    from tensorbored.compat import tensorflow_stub
+
+    return tensorflow_stub
+
+
+@_lazy.lazy_load("tensorbored.compat.tf2")
+def tf2():
+    """Provide the root module of a TF-2.0 API for use within TensorBoard.
+
+    Returns:
+      The root module of a TF-2.0 API, if available.
+
+    Raises:
+      ImportError: if a TF-2.0 API is not available.
+    """
+    # Resolve the lazy `tf` compat API from earlier in this file and try to find
+    # tf.compat.v2. Don't check tf.__version__ since this is not always reliable
+    # if TF was built with tf_api_version!=2.
+    if hasattr(tf, "compat") and hasattr(tf.compat, "v2"):
+        return tf.compat.v2
+    raise ImportError("cannot import tensorflow 2.0 API")

tensorbored/compat/proto/allocation_description_pb2.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/allocation_description.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'\n5tensorbored/compat/proto/allocation_description.proto\x12\x0btensorbored\"\xa3\x01\n\x15\x41llocationDescription\x12\x17\n\x0frequested_bytes\x18\x01 \x01(\x03\x12\x17\n\x0f\x61llocated_bytes\x18\x02 \x01(\x03\x12\x16\n\x0e\x61llocator_name\x18\x03 \x01(\t\x12\x15\n\rallocation_id\x18\x04 \x01(\x03\x12\x1c\n\x14has_single_reference\x18\x05 \x01(\x08\x12\x0b\n\x03ptr\x18\x06 \x01(\x04\x42\x9b\x01\n\x18org.tensorflow.frameworkB\x1b\x41llocationDescriptionProtosP\x01Z]github.com/tensorflow/tensorflow/tensorflow/go/core/framework/allocation_description_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_ALLOCATIONDESCRIPTION = DESCRIPTOR.message_types_by_name['AllocationDescription']
+AllocationDescription = _reflection.GeneratedProtocolMessageType('AllocationDescription', (_message.Message,), {
+  'DESCRIPTOR' : _ALLOCATIONDESCRIPTION,
+  '__module__' : 'tensorbored.compat.proto.allocation_description_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.AllocationDescription)
+  })
+_sym_db.RegisterMessage(AllocationDescription)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\033AllocationDescriptionProtosP\001Z]github.com/tensorflow/tensorflow/tensorflow/go/core/framework/allocation_description_go_proto\370\001\001'
+  _ALLOCATIONDESCRIPTION._serialized_start=71
+  _ALLOCATIONDESCRIPTION._serialized_end=234
+# @@protoc_insertion_point(module_scope)

tensorbored/compat/proto/api_def_pb2.py

@@ -0,0 +1,82 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorbored/compat/proto/api_def.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()
+
+
+from tensorbored.compat.proto import attr_value_pb2 as tensorbored_dot_compat_dot_proto_dot_attr__value__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n&tensorbored/compat/proto/api_def.proto\x12\x0btensorbored\x1a)tensorbored/compat/proto/attr_value.proto\"\xe7\x05\n\x06\x41piDef\x12\x15\n\rgraph_op_name\x18\x01 \x01(\t\x12\x1b\n\x13\x64\x65precation_message\x18\x0c \x01(\t\x12\x1b\n\x13\x64\x65precation_version\x18\r \x01(\x05\x12\x32\n\nvisibility\x18\x02 \x01(\x0e\x32\x1e.tensorbored.ApiDef.Visibility\x12.\n\x08\x65ndpoint\x18\x03 \x03(\x0b\x32\x1c.tensorbored.ApiDef.Endpoint\x12\'\n\x06in_arg\x18\x04 \x03(\x0b\x32\x17.tensorbored.ApiDef.Arg\x12(\n\x07out_arg\x18\x05 \x03(\x0b\x32\x17.tensorbored.ApiDef.Arg\x12\x11\n\targ_order\x18\x0b \x03(\t\x12&\n\x04\x61ttr\x18\x06 \x03(\x0b\x32\x18.tensorbored.ApiDef.Attr\x12\x0f\n\x07summary\x18\x07 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x08 \x01(\t\x12\x1a\n\x12\x64\x65scription_prefix\x18\t \x01(\t\x12\x1a\n\x12\x64\x65scription_suffix\x18\n \x01(\t\x1aI\n\x08\x45ndpoint\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x12\n\ndeprecated\x18\x03 \x01(\x08\x12\x1b\n\x13\x64\x65precation_version\x18\x04 \x01(\x05\x1a;\n\x03\x41rg\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x11\n\trename_to\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x1ak\n\x04\x41ttr\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x11\n\trename_to\x18\x02 \x01(\t\x12-\n\rdefault_value\x18\x03 \x01(\x0b\x32\x16.tensorbored.AttrValue\x12\x13\n\x0b\x64\x65scription\x18\x04 \x01(\t\"G\n\nVisibility\x12\x16\n\x12\x44\x45\x46\x41ULT_VISIBILITY\x10\x00\x12\x0b\n\x07VISIBLE\x10\x01\x12\x08\n\x04SKIP\x10\x02\x12\n\n\x06HIDDEN\x10\x03\"*\n\x07\x41piDefs\x12\x1f\n\x02op\x18\x01 \x03(\x0b\x32\x13.tensorbored.ApiDefB}\n\x18org.tensorflow.frameworkB\x0c\x41piDefProtosP\x01ZNgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/api_def_go_proto\xf8\x01\x01\x62\x06proto3')
+
+
+
+_APIDEF = DESCRIPTOR.message_types_by_name['ApiDef']
+_APIDEF_ENDPOINT = _APIDEF.nested_types_by_name['Endpoint']
+_APIDEF_ARG = _APIDEF.nested_types_by_name['Arg']
+_APIDEF_ATTR = _APIDEF.nested_types_by_name['Attr']
+_APIDEFS = DESCRIPTOR.message_types_by_name['ApiDefs']
+_APIDEF_VISIBILITY = _APIDEF.enum_types_by_name['Visibility']
+ApiDef = _reflection.GeneratedProtocolMessageType('ApiDef', (_message.Message,), {
+
+  'Endpoint' : _reflection.GeneratedProtocolMessageType('Endpoint', (_message.Message,), {
+    'DESCRIPTOR' : _APIDEF_ENDPOINT,
+    '__module__' : 'tensorbored.compat.proto.api_def_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ApiDef.Endpoint)
+    })
+  ,
+
+  'Arg' : _reflection.GeneratedProtocolMessageType('Arg', (_message.Message,), {
+    'DESCRIPTOR' : _APIDEF_ARG,
+    '__module__' : 'tensorbored.compat.proto.api_def_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ApiDef.Arg)
+    })
+  ,
+
+  'Attr' : _reflection.GeneratedProtocolMessageType('Attr', (_message.Message,), {
+    'DESCRIPTOR' : _APIDEF_ATTR,
+    '__module__' : 'tensorbored.compat.proto.api_def_pb2'
+    # @@protoc_insertion_point(class_scope:tensorbored.ApiDef.Attr)
+    })
+  ,
+  'DESCRIPTOR' : _APIDEF,
+  '__module__' : 'tensorbored.compat.proto.api_def_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.ApiDef)
+  })
+_sym_db.RegisterMessage(ApiDef)
+_sym_db.RegisterMessage(ApiDef.Endpoint)
+_sym_db.RegisterMessage(ApiDef.Arg)
+_sym_db.RegisterMessage(ApiDef.Attr)
+
+ApiDefs = _reflection.GeneratedProtocolMessageType('ApiDefs', (_message.Message,), {
+  'DESCRIPTOR' : _APIDEFS,
+  '__module__' : 'tensorbored.compat.proto.api_def_pb2'
+  # @@protoc_insertion_point(class_scope:tensorbored.ApiDefs)
+  })
+_sym_db.RegisterMessage(ApiDefs)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\030org.tensorflow.frameworkB\014ApiDefProtosP\001ZNgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/api_def_go_proto\370\001\001'
+  _APIDEF._serialized_start=99
+  _APIDEF._serialized_end=842
+  _APIDEF_ENDPOINT._serialized_start=526
+  _APIDEF_ENDPOINT._serialized_end=599
+  _APIDEF_ARG._serialized_start=601
+  _APIDEF_ARG._serialized_end=660
+  _APIDEF_ATTR._serialized_start=662
+  _APIDEF_ATTR._serialized_end=769
+  _APIDEF_VISIBILITY._serialized_start=771
+  _APIDEF_VISIBILITY._serialized_end=842
+  _APIDEFS._serialized_start=844
+  _APIDEFS._serialized_end=886
+# @@protoc_insertion_point(module_scope)