Searched defs:on_tpu (Results 1 – 1 of 1) sorted by relevance
1466 def _generate_flush_cache_op(self, num_replicas, on_tpu, argument1568 def _flush_tensor_values_cache(self, tensor_fetches, op_fetches, on_tpu, argument1843 on_tpu=True): argument