def as_default(self): """Enables summary writing within a `with` block.""" if self._resource is None: yield self else: old = context.context().summary_writer_resource context.context().summary_writer_resource = self._resource yield self # Flushes the summary writer in eager mode or in graph functions, but not # in legacy graph mode (you're on your own there). with ops.device("cpu:0"): gen_summary_ops.flush_summary_writer(self._resource) context.context().summary_writer_resource = old
def flush(writer=None, name=None): """Forces summary writer to send any buffered data to storage. This operation blocks until that finishes. Args: writer: The `tf.summary.SummaryWriter` resource to flush. The thread default will be used if this parameter is None. Otherwise a `tf.no_op` is returned. name: A name for the operation (optional). Returns: The created `tf.Operation`. """ if writer is None: writer = context.context().summary_writer if writer is None: return control_flow_ops.no_op() if isinstance(writer, ResourceSummaryWriter): resource = writer._resource # pylint: disable=protected-access else: # Assume we were passed a raw resource tensor. resource = writer with ops.device("cpu:0"): return gen_summary_ops.flush_summary_writer(resource, name=name)
def flush(writer=None, name=None): """Forces summary writer to send any buffered data to storage. This operation blocks until that finishes. Args: writer: The `tf.summary.SummaryWriter` resource to flush. The thread default will be used if this parameter is None. Otherwise a `tf.no_op` is returned. name: A name for the operation (optional). Returns: The created `tf.Operation`. """ if writer is None: writer = context.context().summary_writer if writer is None: return control_flow_ops.no_op() if isinstance(writer, ResourceSummaryWriter): resource = writer._resource # pylint: disable=protected-access else: # Assume we were passed a raw resource tensor. resource = writer with ops.device("cpu:0"): return gen_summary_ops.flush_summary_writer(resource, name=name)
def flush(writer=None, name=None): """Forces summary writer to send any buffered data to storage. This operation blocks until that finishes. Args: writer: The @{tf.contrib.summary.SummaryWriter} resource to flush. The thread default will be used if this parameter is None. Otherwise a @{tf.no_op} is returned. name: A name for the operation (optional). Returns: The created @{tf.Operation}. """ if writer is None: writer = context.context().summary_writer_resource if writer is None: return control_flow_ops.no_op() return gen_summary_ops.flush_summary_writer(writer, name=name)
def flush(writer=None, name=None): """Forces summary writer to send any buffered data to storage. This operation blocks until that finishes. Args: writer: The @{tf.contrib.summary.SummaryWriter} resource to flush. The thread default will be used if this parameter is None. Otherwise a @{tf.no_op} is returned. name: A name for the operation (optional). Returns: The created @{tf.Operation}. """ if writer is None: writer = context.context().summary_writer_resource if writer is None: return control_flow_ops.no_op() return gen_summary_ops.flush_summary_writer(writer, name=name)
def flush(writer=None, name=None): """Forces summary writer to send any buffered data to storage. This operation blocks until that finishes. Args: writer: The `tf.summary.SummaryWriter` to flush. If None, the current default writer will be used instead; if there is no current writer, this returns `tf.no_op`. name: Ignored legacy argument for a name for the operation. Returns: The created `tf.Operation`. """ if writer is None: writer = _summary_state.writer if writer is None: return control_flow_ops.no_op() if isinstance(writer, SummaryWriter): return writer.flush() else: # Legacy fallback in case we were passed a raw resource tensor. with ops.device("cpu:0"): return gen_summary_ops.flush_summary_writer(writer, name=name)
def flush(self): """See `SummaryWriter.flush`.""" with ops.device("cpu:0"): return gen_summary_ops.flush_summary_writer(self._resource)
def flush(self): """See `SummaryWriter.flush`.""" if context.executing_eagerly() and self._closed: return with ops.device("cpu:0"): return gen_summary_ops.flush_summary_writer(self._resource)