| # 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. |
| """ |
| # 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 * |
| |
| # Data |
| from tensorflow.python import data |
| |
| # 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 * # pylint: disable=redefined-builtin |
| from tensorflow.python.ops.random_crop_ops import * |
| |
| # Function |
| from tensorflow.core.function.trace_type 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.experimental.rpc.rpc_ops import * |
| from tensorflow.python.distribute.merge_call_interim import * |
| from tensorflow.python.distribute.multi_process_runner import * |
| from tensorflow.python.distribute.multi_worker_test_base import * |
| from tensorflow.python.distribute.sharded_variable 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 * |
| from tensorflow.python.distribute.failure_handling.failure_handling import * |
| from tensorflow.python.distribute.failure_handling.preemption_watcher import * |
| |
| from tensorflow.python.ops.numpy_ops import np_random |
| from tensorflow.python.ops.numpy_ops import np_utils |
| from tensorflow.python.ops.numpy_ops import np_array_ops |
| from tensorflow.python.ops.numpy_ops import np_arrays |
| from tensorflow.python.ops.numpy_ops import np_config |
| from tensorflow.python.ops.numpy_ops import np_dtypes |
| from tensorflow.python.ops.numpy_ops import np_math_ops |
| |
| 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=['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 |