| 886 | self._data = {} |
| 887 | |
| 888 | def begin(self, max_steps=None): |
| 889 | super(GraphDump, self).begin(max_steps=max_steps) |
| 890 | self._tensors = [] |
| 891 | graph = ops.get_default_graph() |
| 892 | graph_def = graph.as_graph_def() |
| 893 | for node in graph_def.node: |
| 894 | if node.op in self._ignore_ops: |
| 895 | continue |
| 896 | logging.info("op=%s name=%s.", node.op, node.name) |
| 897 | try: |
| 898 | self._tensors.append(graph.get_tensor_by_name(node.name + ":0")) |
| 899 | except KeyError: |
| 900 | pass |
| 901 | |
| 902 | def step_begin(self, step): |
| 903 | super(GraphDump, self).step_begin(step) |