Home | History | Annotate | Download | only in tpu
      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 """Stub file to maintain backwards compatibility."""
     16 
     17 from __future__ import absolute_import
     18 from __future__ import division
     19 from __future__ import print_function
     20 
     21 # pylint: disable=wildcard-import,unused-import,redefined-builtin
     22 from tensorflow.python.tpu.tpu_estimator import *
     23 # used by tests
     24 from tensorflow.python.tpu.tpu_estimator import _clone_export_output_with_tensors
     25 from tensorflow.python.tpu.tpu_estimator import _create_global_step
     26 from tensorflow.python.tpu.tpu_estimator import _export_output_to_tensors
     27 from tensorflow.python.tpu.tpu_estimator import _get_scaffold
     28 from tensorflow.python.tpu.tpu_estimator import _Inputs
     29 from tensorflow.python.tpu.tpu_estimator import _ITERATIONS_PER_LOOP_VAR
     30 from tensorflow.python.tpu.tpu_estimator import _TPU_ENQUEUE_OPS
     31 from tensorflow.python.tpu.tpu_estimator import _TPU_ESTIMATOR
     32 from tensorflow.python.tpu.tpu_estimator import _TPU_TRAIN_OP
     33 # pylint: enable=wildcard-import,unused-import,redefined-builtin
     34