(self)
| 1022 | return tensor_trace_order |
| 1023 | |
| 1024 | def _create_host_call(self): |
| 1025 | return self._parameters.trace_mode in ( |
| 1026 | tensor_tracer_flags.TRACE_MODE_SUMMARY, |
| 1027 | tensor_tracer_flags.TRACE_MODE_FULL_TENSOR_SUMMARY) |
| 1028 | |
| 1029 | |
| 1030 | def _generate_flush_cache_op(self, num_replicas, on_tpu): |