Searched refs:on_tpu (Results 1 – 1 of 1) sorted by relevance
1243 def _generate_flush_cache_op(self, graph, start_replica, on_tpu): argument1284 if on_tpu:1319 on_tpu): argument1338 flush_op = self._generate_flush_cache_op(graph, start_replica, on_tpu)1413 on_tpu=True): argument1493 if on_tpu:1501 elif on_tpu:1530 on_tpu=on_tpu)1592 on_tpu=True)1634 on_tpu=False)