def WriteDebuggedGraph(self, debugged_graph): """Write a DebuggedGraph proto with the writer. Args: debugged_graph: A DebuggedGraph proto, describing the details of a TensorFlow Graph that has completed its construction. """ debug_event = debug_event_pb2.DebugEvent(debugged_graph=debugged_graph) self._EnsureTimestampAdded(debug_event) _pywrap_debug_events_writer.WriteDebuggedGraph(self._dump_root, debug_event)
def WriteExecution(self, execution): """Write a Execution proto with the writer. Args: execution: An Execution proto, describing a TensorFlow op or graph execution event. """ debug_event = debug_event_pb2.DebugEvent(execution=execution) self._EnsureTimestampAdded(debug_event) _pywrap_debug_events_writer.WriteExecution(self._dump_root, debug_event)
def ReadDebugEvents(filename): reader = tf_record.tf_record_iterator(filename) debug_events = [] try: while True: debug_event = debug_event_pb2.DebugEvent() debug_event.ParseFromString(next(reader)) debug_events.append(debug_event) except StopIteration: return debug_events
def WriteSourceFile(self, source_file): """Write a SourceFile proto with the writer. Args: source_file: A SourceFile proto, describing the content of a source file involved in the execution of the debugged TensorFlow program. """ # TODO(cais): Explore performance optimization that avoids memcpy. debug_event = debug_event_pb2.DebugEvent(source_file=source_file) self._EnsureTimestampAdded(debug_event) _pywrap_debug_events_writer.WriteSourceFile(self._dump_root, debug_event)
def WriteGraphOpCreation(self, graph_op_creation): """Write a GraphOpCreation proto with the writer. Args: graph_op_creation: A GraphOpCreation proto, describing the details of the creation of an op inside a TensorFlow Graph. """ debug_event = debug_event_pb2.DebugEvent( graph_op_creation=graph_op_creation) self._EnsureTimestampAdded(debug_event) _pywrap_debug_events_writer.WriteGraphOpCreation(self._dump_root, debug_event)
def WriteStackFrameWithId(self, stack_frame_with_id): """Write a StackFrameWithId proto with the writer. Args: stack_frame_with_id: A StackFrameWithId proto, describing the content a stack frame involved in the execution of the debugged TensorFlow program. """ debug_event = debug_event_pb2.DebugEvent( stack_frame_with_id=stack_frame_with_id) self._EnsureTimestampAdded(debug_event) _pywrap_debug_events_writer.WriteStackFrameWithId(self._dump_root, debug_event)
def WriteGraphExecutionTrace(self, graph_execution_trace): """Write a GraphExecutionTrace proto with the writer. Args: graph_execution_trace: A GraphExecutionTrace proto, concerning the value of an intermediate tensor or a list of intermediate tensors that are computed during the graph's execution. """ debug_event = debug_event_pb2.DebugEvent( graph_execution_trace=graph_execution_trace) self._EnsureTimestampAdded(debug_event) _pywrap_debug_events_writer.WriteGraphExecutionTrace( self._dump_root, debug_event)