32f35aabce
This change exports the following class symbols, and adds relevant documentation and example code to tf.distribute.experimental.ParameterServerStrategy tf.distribute.experimental.coordinator.ClusterCoordinator tf.distribute.experimental.coordinator.PerWorkerValues tf.distribute.experimental.coordinator.RemoteValue PiperOrigin-RevId: 338151262 Change-Id: If2d1c513d30a999c728cecc2e73b75adda1948c2
97 lines
3.9 KiB
Python
97 lines
3.9 KiB
Python
# Copyright 2020 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.
|
|
# ==============================================================================
|
|
"""Imports modules that should be scanned during API generation.
|
|
|
|
This file should eventually contain everything we need to scan looking for
|
|
tf_export decorators.
|
|
"""
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
|
|
# go/tf-wildcard-import
|
|
# pylint: disable=wildcard-import,g-bad-import-order,g-import-not-at-top
|
|
# pylint: disable=unused-import,g-importing-member
|
|
|
|
# Protocol buffers
|
|
from tensorflow.core.framework.graph_pb2 import *
|
|
from tensorflow.core.framework.node_def_pb2 import *
|
|
from tensorflow.core.framework.summary_pb2 import *
|
|
from tensorflow.core.framework.attr_value_pb2 import *
|
|
from tensorflow.core.protobuf.meta_graph_pb2 import TensorInfo
|
|
from tensorflow.core.protobuf.meta_graph_pb2 import MetaGraphDef
|
|
from tensorflow.core.protobuf.config_pb2 import *
|
|
from tensorflow.core.util.event_pb2 import *
|
|
|
|
# Framework
|
|
from tensorflow.python.framework.framework_lib import * # pylint: disable=redefined-builtin
|
|
from tensorflow.python.framework.versions import *
|
|
from tensorflow.python.framework import config
|
|
from tensorflow.python.framework import errors
|
|
from tensorflow.python.framework import graph_util
|
|
|
|
# Session
|
|
from tensorflow.python.client.client_lib import *
|
|
|
|
# Ops
|
|
from tensorflow.python.ops.standard_ops import *
|
|
|
|
# Namespaces
|
|
from tensorflow.python.ops import initializers_ns as initializers
|
|
|
|
from tensorflow.python.util.tf_export import tf_export
|
|
|
|
# _internal APIs
|
|
from tensorflow.python.distribute.combinations import generate
|
|
from tensorflow.python.distribute.multi_process_runner import *
|
|
from tensorflow.python.distribute.multi_worker_test_base import *
|
|
from tensorflow.python.distribute.strategy_combinations import *
|
|
from tensorflow.python.framework.combinations import *
|
|
from tensorflow.python.framework.composite_tensor import *
|
|
from tensorflow.python.framework.test_combinations import *
|
|
from tensorflow.python.util.tf_decorator import make_decorator
|
|
from tensorflow.python.util.tf_decorator import unwrap
|
|
|
|
from tensorflow.python.distribute.parameter_server_strategy_v2 import *
|
|
from tensorflow.python.distribute.coordinator.cluster_coordinator import *
|
|
|
|
tf_export('__internal__.decorator.make_decorator', v1=[])(make_decorator)
|
|
tf_export('__internal__.decorator.unwrap', v1=[])(unwrap)
|
|
|
|
|
|
# Export protos
|
|
# pylint: disable=undefined-variable
|
|
tf_export(v1=['AttrValue'])(AttrValue)
|
|
tf_export(v1=['ConfigProto'])(ConfigProto)
|
|
tf_export(v1=['Event', 'summary.Event'])(Event)
|
|
tf_export(v1=['GPUOptions'])(GPUOptions)
|
|
tf_export(v1=['GraphDef'])(GraphDef)
|
|
tf_export(v1=['GraphOptions'])(GraphOptions)
|
|
tf_export(v1=['HistogramProto'])(HistogramProto)
|
|
tf_export(v1=['LogMessage'])(LogMessage)
|
|
tf_export(v1=['MetaGraphDef'])(MetaGraphDef)
|
|
tf_export(v1=['NameAttrList'])(NameAttrList)
|
|
tf_export(v1=['NodeDef'])(NodeDef)
|
|
tf_export(v1=['OptimizerOptions'])(OptimizerOptions)
|
|
tf_export(v1=['RunMetadata'])(RunMetadata)
|
|
tf_export(v1=['RunOptions'])(RunOptions)
|
|
tf_export(v1=['SessionLog', 'summary.SessionLog'])(SessionLog)
|
|
tf_export(v1=['Summary', 'summary.Summary'])(Summary)
|
|
tf_export(v1=['summary.SummaryDescription'])(SummaryDescription)
|
|
tf_export(v1=['SummaryMetadata'])(SummaryMetadata)
|
|
tf_export(v1=['summary.TaggedRunMetadata'])(TaggedRunMetadata)
|
|
tf_export(v1=['TensorInfo'])(TensorInfo)
|
|
# pylint: enable=undefined-variable
|