Searched refs:on_tpu (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer.py | 1369 def _generate_flush_cache_op(self, num_replicas, on_tpu, tensor_trace_order): argument
1423 if on_tpu and not self._parameters.collect_summary_per_core:
1442 if on_tpu:
1468 def _flush_tensor_values_cache(self, tensor_fetches, op_fetches, on_tpu, argument
1486 self._tt_config.num_replicas, on_tpu, tensor_trace_order)
1729 on_tpu=True): argument
1822 if on_tpu:
1851 if on_tpu:
1896 if not on_tpu:
1930 processed_t_fetches, op_fetches, on_tpu=on_tpu,
[all …]
|