Lines Matching refs:node_name
64 node_name=None,
74 node_name: (str or None) name of the node involved (if any). If None,
92 if node_name:
96 "node_info -a -d -t %s" % node_name,
101 "print_tensor %s" % node_name,
106 "list_inputs -c -r %s" % node_name,
111 "list_outputs -c -r %s" % node_name,
223 "node_name",
252 "node_name",
283 "node_name",
381 1) in the simplest case, use <node_name>:<output_slot>, e.g.,
385 <node_name>:<output_slot>:<debug_op>, e.g.,
389 <device_name>:<node_name>:<output_slot>[:<debug_op>], e.g.,
520 _add_main_menu(output, node_name=None, enable_list_tensors=False)
542 if node_name_regex and not node_name_regex.match(dump.node_name):
546 op_type = self._debug_dump.node_op_type(dump.node_name)
552 dumped_tensor_name = "%s:%d" % (dump.node_name, dump.output_slot)
553 op_type = self._debug_dump.node_op_type(dump.node_name)
579 _add_main_menu(output, node_name=None, enable_list_tensors=False)
615 op_type = self._debug_dump.node_op_type(dump.node_name)
649 key=lambda x: self._debug_dump.node_op_type(x.node_name))
654 key=lambda x: "%s:%d" % (x.node_name, x.output_slot))
745 node_name, unused_slot = debug_graphs.parse_node_or_tensor_name(
746 parsed.node_name)
748 if not self._debug_dump.node_exists(node_name):
750 "There is no node named \"%s\" in the partition graphs" % node_name)
753 node_name=None,
769 lines = ["Node %s" % node_name]
771 0: [(len(lines[-1]) - len(node_name), len(lines[-1]), "bold")]
774 lines.append(" Op: %s" % self._debug_dump.node_op_type(node_name))
775 lines.append(" Device: %s" % self._debug_dump.node_device(node_name))
781 self._debug_dump.node_inputs(node_name))
783 self._debug_dump.node_inputs(node_name, is_control=True))
788 self._debug_dump.node_recipients(node_name))
790 self._debug_dump.node_recipients(node_name, is_control=True))
795 output.extend(self._list_node_attributes(node_name))
799 output.extend(self._list_node_dumps(node_name))
802 output.extend(self._render_node_traceback(node_name))
804 _add_main_menu(output, node_name=node_name, enable_node_info=False)
816 return [node_name for node_name in node_names
818 debug_graphs.get_node_name(node_name)) not in
821 def _render_node_traceback(self, node_name):
825 node_name: (str) name of the node.
835 node_stack = self._debug_dump.node_traceback(node_name)
881 parsed.node_name,
887 node_name = debug_graphs.get_node_name(parsed.node_name)
888 _add_main_menu(output, node_name=node_name, enable_list_inputs=False)
918 node_name, output_slot = debug_graphs.parse_node_or_tensor_name(tensor_name)
920 not self._debug_dump.node_exists(node_name)):
922 "Node \"%s\" does not exist in partition graphs" % node_name)
925 node_name=None,
930 watch_keys = self._debug_dump.debug_watch_keys(node_name)
947 (node_name, len(output_slots)),
948 "Please specify the output slot: %s:x." % node_name
953 node_name=node_name,
988 _add_main_menu(output, node_name=node_name, enable_print_tensor=False)
1030 _add_main_menu(output, node_name=node_name, enable_print_tensor=False)
1059 parsed.node_name,
1065 node_name = debug_graphs.get_node_name(parsed.node_name)
1066 _add_main_menu(output, node_name=node_name, enable_list_outputs=False)
1149 _add_main_menu(output, node_name=None)
1252 _add_main_menu(output, node_name=None)
1257 node_name,
1269 node_name: The name of the node in question, as a str.
1296 node_name, _ = debug_graphs.parse_node_or_tensor_name(node_name)
1299 if not self._debug_dump.node_exists(node_name):
1301 "There is no node named \"%s\" in the partition graphs" % node_name)
1313 line = "%s node \"%s\"" % (type_str, node_name)
1314 font_attr_segs[0] = [(len(line) - 1 - len(node_name), len(line) - 1, "bold")
1323 node_name,
1349 node_name,
1360 node_name, and append these inputs (or output recipients) to a list of text
1368 node_name: Name of the node, as a str. This arg is updated during the
1387 copy.copy(tracker(node_name, is_control=False)))
1393 sorted(tracker(node_name, is_control=True)))
1506 def _list_node_attributes(self, node_name):
1510 node_name: Name of the node of which the attributes are to be listed.
1520 attrs = self._debug_dump.node_attributes(node_name)
1529 def _list_node_dumps(self, node_name):
1533 node_name: Name of the node of which the attributes are to be listed.
1542 watch_keys = self._debug_dump.debug_watch_keys(node_name)
1552 command = "pt %s:%d -n %d" % (node_name, datum.output_slot, dump_count)
1637 dumped_tensor_names.append("%s:%d" % (datum.node_name, datum.output_slot))
