From 4204c5f8565c9b5f7027c8d1b7ba7897754a297e Mon Sep 17 00:00:00 2001 From: Ruoxin Sang Date: Fri, 8 May 2020 17:09:34 -0700 Subject: [PATCH] Remove `_enable_legacy_iterators` flag in TPUStrategy. PiperOrigin-RevId: 310656609 Change-Id: I0d4eaa8139689df1b6f644c87b1cd4028831dd52 --- tensorflow/python/distribute/tpu_strategy.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/tensorflow/python/distribute/tpu_strategy.py b/tensorflow/python/distribute/tpu_strategy.py index 82a4a803510..6e51b84a1d1 100644 --- a/tensorflow/python/distribute/tpu_strategy.py +++ b/tensorflow/python/distribute/tpu_strategy.py @@ -140,10 +140,6 @@ class TPUStrategy(distribute_lib.Strategy): distribute_lib.distribution_strategy_replica_gauge.get_cell( "num_replicas_per_worker").set(self.extended.num_replicas_per_host) - # TODO(b/155193424): Enable OwnedMultiDeviceIterator on TPU Pod. - if self.extended.num_hosts > 1: - self._enable_legacy_iterators = True - # TODO(cjfj): Modify `_call_for_each_replica` in `TPUExtended` such that this # can use the default implementation. # This implementation runs a single step. It does not use infeed or outfeed.