Migrate references and remove legacy target tpu:tpu_estimator.

PiperOrigin-RevId: 582462371
tensorflow · Nov 16, 2023 · 207c70c · 207c70c
1 parent fbf7b1e
commit 207c70c
Show file tree

Hide file tree

Showing 77 changed files with 81 additions and 81 deletions.
diff --git a/examples/mnist.py b/examples/mnist.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/examples/mnist_dataset.py b/examples/mnist_dataset.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/examples/toy_model_tpu.py b/examples/toy_model_tpu.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -27,8 +27,8 @@
 from tensorflow.python.data.ops.dataset_ops import Dataset
 from tensorflow.python.platform import flags
 from tensorflow.python.platform import tf_logging as logging
-from tensorflow.python.tpu import tpu_config  # pylint: disable=g-direct-tensorflow-import
-from tensorflow.python.tpu import tpu_estimator  # pylint: disable=g-direct-tensorflow-import
+from tensorflow_estimator.python.estimator.tpu import tpu_config
+from tensorflow_estimator.python.estimator.tpu import tpu_estimator
 from tensorflow_estimator.python.estimator import estimator as estimator_lib
 
 FLAGS = flags.FLAGS

diff --git a/mesh_tensorflow/__init__.py b/mesh_tensorflow/__init__.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/auto_mtf/__init__.py b/mesh_tensorflow/auto_mtf/__init__.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/auto_mtf/api.py b/mesh_tensorflow/auto_mtf/api.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/auto_mtf/api_test.py b/mesh_tensorflow/auto_mtf/api_test.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/auto_mtf/graph_interface.py b/mesh_tensorflow/auto_mtf/graph_interface.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/auto_mtf/graph_interface_test.py b/mesh_tensorflow/auto_mtf/graph_interface_test.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/auto_mtf/layout_optimizer.py b/mesh_tensorflow/auto_mtf/layout_optimizer.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/auto_mtf/layout_optimizer_test.py b/mesh_tensorflow/auto_mtf/layout_optimizer_test.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/auto_mtf/memory_estimator.py b/mesh_tensorflow/auto_mtf/memory_estimator.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/auto_mtf/memory_estimator_test.py b/mesh_tensorflow/auto_mtf/memory_estimator_test.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/auto_mtf/print_cp_model_solution.py b/mesh_tensorflow/auto_mtf/print_cp_model_solution.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/auto_mtf/scheduler.py b/mesh_tensorflow/auto_mtf/scheduler.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/auto_mtf/scheduler_test.py b/mesh_tensorflow/auto_mtf/scheduler_test.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/auto_mtf/valid_layouts.py b/mesh_tensorflow/auto_mtf/valid_layouts.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/auto_mtf/valid_layouts_test.py b/mesh_tensorflow/auto_mtf/valid_layouts_test.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/beam_search.py b/mesh_tensorflow/beam_search.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/bert/__init__.py b/mesh_tensorflow/bert/__init__.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/bert/bert.py b/mesh_tensorflow/bert/bert.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/bert/optimization.py b/mesh_tensorflow/bert/optimization.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/bert/run_classifier.py b/mesh_tensorflow/bert/run_classifier.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/bert/run_pretraining.py b/mesh_tensorflow/bert/run_pretraining.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/bert/run_squad.py b/mesh_tensorflow/bert/run_squad.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/bert/tokenization.py b/mesh_tensorflow/bert/tokenization.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/experimental/__init__.py b/mesh_tensorflow/experimental/__init__.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/experimental/data_aug_lib.py b/mesh_tensorflow/experimental/data_aug_lib.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/experimental/data_aug_lib_test.py b/mesh_tensorflow/experimental/data_aug_lib_test.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/experimental/input_reader.py b/mesh_tensorflow/experimental/input_reader.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/experimental/input_reader_test.py b/mesh_tensorflow/experimental/input_reader_test.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/experimental/model_executor.py b/mesh_tensorflow/experimental/model_executor.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/experimental/offline_data_aug.py b/mesh_tensorflow/experimental/offline_data_aug.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/experimental/unet.py b/mesh_tensorflow/experimental/unet.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/import_test.py b/mesh_tensorflow/import_test.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/layers.py b/mesh_tensorflow/layers.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/layers_test.py b/mesh_tensorflow/layers_test.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/ops.py b/mesh_tensorflow/ops.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/ops_test.py b/mesh_tensorflow/ops_test.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/ops_with_redefined_builtins.py b/mesh_tensorflow/ops_with_redefined_builtins.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/optimize.py b/mesh_tensorflow/optimize.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/placement_mesh_impl.py b/mesh_tensorflow/placement_mesh_impl.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/simd_mesh_impl.py b/mesh_tensorflow/simd_mesh_impl.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/simd_mesh_impl_test.py b/mesh_tensorflow/simd_mesh_impl_test.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/test_utils.py b/mesh_tensorflow/test_utils.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/test_utils_test.py b/mesh_tensorflow/test_utils_test.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/tpu_variables.py b/mesh_tensorflow/tpu_variables.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/transformer/__init__.py b/mesh_tensorflow/transformer/__init__.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/transformer/adaptive_softmax.py b/mesh_tensorflow/transformer/adaptive_softmax.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/transformer/adaptive_softmax_test.py b/mesh_tensorflow/transformer/adaptive_softmax_test.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/transformer/attention.py b/mesh_tensorflow/transformer/attention.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/transformer/dataset.py b/mesh_tensorflow/transformer/dataset.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/transformer/dataset_test.py b/mesh_tensorflow/transformer/dataset_test.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/transformer/evolved_transformer.py b/mesh_tensorflow/transformer/evolved_transformer.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

diff --git a/mesh_tensorflow/transformer/fixup_layers.py b/mesh_tensorflow/transformer/fixup_layers.py
@@ -1,5 +1,5 @@
 # coding=utf-8
-# Copyright 2022 The Mesh TensorFlow Authors.
+# Copyright 2023 The Mesh TensorFlow Authors.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.