Internal change

PiperOrigin-RevId: 501022379

Author: fyangf

official/vision/modeling/layers/edgetpu.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 """EdgeTPU oriented layers and tools."""
-from typing import List, Optional, Sequence, Union, Iterable
+from typing import List, Optional, Union, Iterable, Sequence
 
 import numpy as np
 import tensorflow as tf
@@ -101,7 +101,7 @@ def _same(x):
 
 
 def shard_tensors(
-    axis: int, block_size: int, tensors: Sequence[tf.Tensor]
+    axis: int, block_size: int, tensors: 'Sequence[tf.Tensor]'
 ) -> Union[List[Sequence[tf.Tensor]], 'Iterable[Sequence[tf.Tensor]]']:
   """Consistently splits multiple tensors sharding-style.