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""" 20from __future__ import absolute_import 21from __future__ import division 22from __future__ import print_function 23 24# go/tf-wildcard-import 25# pylint: disable=wildcard-import,g-bad-import-order,g-import-not-at-top 26# pylint: disable=unused-import,g-importing-member 27 28# Protocol buffers 29from tensorflow.core.framework.graph_pb2 import * 30from tensorflow.core.framework.node_def_pb2 import * 31from tensorflow.core.framework.summary_pb2 import * 32from tensorflow.core.framework.attr_value_pb2 import * 33from tensorflow.core.protobuf.meta_graph_pb2 import TensorInfo 34from tensorflow.core.protobuf.meta_graph_pb2 import MetaGraphDef 35from tensorflow.core.protobuf.config_pb2 import * 36from tensorflow.core.util.event_pb2 import * 37 38# Framework 39from tensorflow.python.framework.framework_lib import * # pylint: disable=redefined-builtin 40from tensorflow.python.framework.versions import * 41from tensorflow.python.framework import config 42from tensorflow.python.framework import errors 43from tensorflow.python.framework import graph_util 44 45# Session 46from tensorflow.python.client.client_lib import * 47 48# Ops 49from tensorflow.python.ops.standard_ops import * 50 51# Namespaces 52from tensorflow.python.ops import initializers_ns as initializers 53 54from tensorflow.python.util.tf_export import tf_export 55 56# _internal APIs 57from tensorflow.python.distribute.combinations import generate 58from tensorflow.python.distribute.multi_process_runner import * 59from tensorflow.python.distribute.multi_worker_test_base import * 60from tensorflow.python.distribute.strategy_combinations import * 61from tensorflow.python.framework.combinations import * 62from tensorflow.python.framework.composite_tensor import * 63from tensorflow.python.framework.test_combinations import * 64from tensorflow.python.util.tf_decorator import make_decorator 65from tensorflow.python.util.tf_decorator import unwrap 66 67from tensorflow.python.distribute.parameter_server_strategy_v2 import * 68from tensorflow.python.distribute.coordinator.cluster_coordinator import * 69 70tf_export('__internal__.decorator.make_decorator', v1=[])(make_decorator) 71tf_export('__internal__.decorator.unwrap', v1=[])(unwrap) 72 73 74# Export protos 75# pylint: disable=undefined-variable 76tf_export(v1=['AttrValue'])(AttrValue) 77tf_export(v1=['ConfigProto'])(ConfigProto) 78tf_export(v1=['Event', 'summary.Event'])(Event) 79tf_export(v1=['GPUOptions'])(GPUOptions) 80tf_export(v1=['GraphDef'])(GraphDef) 81tf_export(v1=['GraphOptions'])(GraphOptions) 82tf_export(v1=['HistogramProto'])(HistogramProto) 83tf_export(v1=['LogMessage'])(LogMessage) 84tf_export(v1=['MetaGraphDef'])(MetaGraphDef) 85tf_export(v1=['NameAttrList'])(NameAttrList) 86tf_export(v1=['NodeDef'])(NodeDef) 87tf_export(v1=['OptimizerOptions'])(OptimizerOptions) 88tf_export(v1=['RunMetadata'])(RunMetadata) 89tf_export(v1=['RunOptions'])(RunOptions) 90tf_export(v1=['SessionLog', 'summary.SessionLog'])(SessionLog) 91tf_export(v1=['Summary', 'summary.Summary'])(Summary) 92tf_export(v1=['summary.SummaryDescription'])(SummaryDescription) 93tf_export(v1=['SummaryMetadata'])(SummaryMetadata) 94tf_export(v1=['summary.TaggedRunMetadata'])(TaggedRunMetadata) 95tf_export(v1=['TensorInfo'])(TensorInfo) 96# pylint: enable=undefined-variable 97