1# Copyright 2020 The TensorFlow Authors. All Rights Reserved. 2# 3# Licensed under the Apache License, Version 2.0 (the "License"); 4# you may not use this file except in compliance with the License. 5# You may obtain a copy of the License at 6# 7# http://www.apache.org/licenses/LICENSE-2.0 8# 9# Unless required by applicable law or agreed to in writing, software 10# distributed under the License is distributed on an "AS IS" BASIS, 11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12# See the License for the specific language governing permissions and 13# limitations under the License. 14# ============================================================================== 15"""Imports modules that should be scanned during API generation. 16 17This file should eventually contain everything we need to scan looking for 18tf_export decorators. 19""" 20# go/tf-wildcard-import 21# pylint: disable=wildcard-import,g-bad-import-order,g-import-not-at-top 22# pylint: disable=unused-import,g-importing-member 23 24# Protocol buffers 25from tensorflow.core.framework.graph_pb2 import * 26from tensorflow.core.framework.node_def_pb2 import * 27from tensorflow.core.framework.summary_pb2 import * 28from tensorflow.core.framework.attr_value_pb2 import * 29from tensorflow.core.protobuf.meta_graph_pb2 import TensorInfo 30from tensorflow.core.protobuf.meta_graph_pb2 import MetaGraphDef 31from tensorflow.core.protobuf.config_pb2 import * 32from tensorflow.core.util.event_pb2 import * 33 34# Framework 35from tensorflow.python.framework.framework_lib import * # pylint: disable=redefined-builtin 36from tensorflow.python.framework.versions import * 37from tensorflow.python.framework import config 38from tensorflow.python.framework import errors 39from tensorflow.python.framework import graph_util 40 41# Session 42from tensorflow.python.client.client_lib import * 43 44# Ops 45from tensorflow.python.ops.standard_ops import * # pylint: disable=redefined-builtin 46 47# Function 48from tensorflow.core.function.trace_type import * 49 50# Namespaces 51from tensorflow.python.ops import initializers_ns as initializers 52 53from tensorflow.python.util.tf_export import tf_export 54 55# _internal APIs 56from tensorflow.python.distribute.combinations import generate 57from tensorflow.python.distribute.experimental.rpc.rpc_ops import * 58from tensorflow.python.distribute.merge_call_interim import * 59from tensorflow.python.distribute.multi_process_runner import * 60from tensorflow.python.distribute.multi_worker_test_base import * 61from tensorflow.python.distribute.sharded_variable import * 62from tensorflow.python.distribute.strategy_combinations import * 63from tensorflow.python.framework.combinations import * 64from tensorflow.python.framework.composite_tensor import * 65from tensorflow.python.framework.test_combinations import * 66from tensorflow.python.util.tf_decorator import make_decorator 67from tensorflow.python.util.tf_decorator import unwrap 68 69from tensorflow.python.distribute.parameter_server_strategy_v2 import * 70from tensorflow.python.distribute.coordinator.cluster_coordinator import * 71from tensorflow.python.distribute.failure_handling.failure_handling import * 72 73tf_export('__internal__.decorator.make_decorator', v1=[])(make_decorator) 74tf_export('__internal__.decorator.unwrap', v1=[])(unwrap) 75 76 77# Export protos 78# pylint: disable=undefined-variable 79tf_export(v1=['AttrValue'])(AttrValue) 80tf_export(v1=['ConfigProto'])(ConfigProto) 81tf_export(v1=['Event', 'summary.Event'])(Event) 82tf_export(v1=['GPUOptions'])(GPUOptions) 83tf_export(v1=['GraphOptions'])(GraphOptions) 84tf_export(v1=['HistogramProto'])(HistogramProto) 85tf_export(v1=['LogMessage'])(LogMessage) 86tf_export(v1=['MetaGraphDef'])(MetaGraphDef) 87tf_export(v1=['NameAttrList'])(NameAttrList) 88tf_export(v1=['NodeDef'])(NodeDef) 89tf_export(v1=['OptimizerOptions'])(OptimizerOptions) 90tf_export(v1=['RunMetadata'])(RunMetadata) 91tf_export(v1=['RunOptions'])(RunOptions) 92tf_export(v1=['SessionLog', 'summary.SessionLog'])(SessionLog) 93tf_export(v1=['Summary', 'summary.Summary'])(Summary) 94tf_export(v1=['summary.SummaryDescription'])(SummaryDescription) 95tf_export(v1=['SummaryMetadata'])(SummaryMetadata) 96tf_export(v1=['summary.TaggedRunMetadata'])(TaggedRunMetadata) 97tf_export(v1=['TensorInfo'])(TensorInfo) 98# pylint: enable=undefined-variable 99