[Clean up] Move TF1 only TPU utils to r1 folder.

PiperOrigin-RevId: 279798675

official/utils/accelerator/tpu_test.py → official/r1/utils/tpu_test.py

@@ -17,7 +17,7 @@
 import numpy as np
 import tensorflow as tf
 
-from official.utils.accelerator import tpu as tpu_utils
+from official.r1.utils import tpu as tpu_utils
 
 
 TEST_CASES = [

official/transformer/model/embedding_layer.py

@@ -21,7 +21,7 @@ from __future__ import print_function
 import tensorflow as tf  # pylint: disable=g-bad-import-order
 
 from official.transformer.model import model_utils
-from official.utils.accelerator import tpu as tpu_utils
+from official.r1.utils import tpu as tpu_utils
 
 
 class EmbeddingSharedWeights(tf.layers.Layer):

official/transformer/transformer_main.py

@@ -33,6 +33,7 @@ import tensorflow as tf
 # pylint: enable=g-bad-import-order
 
 from official.r1.utils import export
+from official.r1.utils import tpu as tpu_util
 from official.transformer import compute_bleu
 from official.transformer import translate
 from official.transformer.model import model_params
@@ -41,7 +42,6 @@ from official.transformer.utils import dataset
 from official.transformer.utils import metrics
 from official.transformer.utils import schedule
 from official.transformer.utils import tokenizer
-from official.utils.accelerator import tpu as tpu_util
 from official.utils.flags import core as flags_core
 from official.utils.logs import hooks_helper
 from official.utils.logs import logger