Searched defs:on_tpu (Results 1 – 1 of 1) sorted by relevance
1369 def _generate_flush_cache_op(self, num_replicas, on_tpu, tensor_trace_order): argument1468 def _flush_tensor_values_cache(self, tensor_fetches, op_fetches, on_tpu, argument1729 on_tpu=True): argument