1 # Copyright 2019 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 """Modules that need to be exported to the API. 16 17 List TPU modules that aren't included elsewhere here so that they can be scanned 18 for tf_export decorations. 19 """ 20 21 # pylint: disable=unused-import 22 from tensorflow.python.tpu import bfloat16 23 from tensorflow.python.tpu import feature_column_v2 24 from tensorflow.python.tpu import tpu 25 from tensorflow.python.tpu import tpu_embedding 26 from tensorflow.python.tpu import tpu_embedding_for_serving 27 from tensorflow.python.tpu import tpu_embedding_v1 28 from tensorflow.python.tpu import tpu_embedding_v2 29 from tensorflow.python.tpu import tpu_embedding_v2_utils 30 from tensorflow.python.tpu import tpu_hardware_feature 31 from tensorflow.python.tpu import tpu_optimizer 32 # pylint: enable=unused-import 33