|
1 | 1 | # coding=utf-8 |
2 | | -# Copyright2022 The Mesh TensorFlow Authors. |
| 2 | +# Copyright2023 The Mesh TensorFlow Authors. |
3 | 3 | # |
4 | 4 | # Licensed under the Apache License, Version 2.0 (the "License"); |
5 | 5 | # you may not use this file except in compliance with the License. |
|
27 | 27 | fromtensorflow.python.data.ops.dataset_opsimportDataset |
28 | 28 | fromtensorflow.python.platformimportflags |
29 | 29 | fromtensorflow.python.platformimporttf_loggingaslogging |
30 | | -fromtensorflow.python.tpuimporttpu_config# pylint: disable=g-direct-tensorflow-import |
31 | | -fromtensorflow.python.tpuimporttpu_estimator# pylint: disable=g-direct-tensorflow-import |
32 | 30 | fromtensorflow_estimator.python.estimatorimportestimatorasestimator_lib |
| 31 | +fromtensorflow_estimator.python.estimator.tpuimporttpu_config# pylint: disable=g-deprecated-tf-checker |
| 32 | +fromtensorflow_estimator.python.estimator.tpuimporttpu_estimator# pylint: disable=g-deprecated-tf-checker |
33 | 33 |
|
34 | 34 | FLAGS=flags.FLAGS |
35 | 35 |
|
|