Searched refs:debug_graph (Results 1 – 1 of 1) sorted by relevance
793 debug_graph = debug_graphs.DebugGraph(partition_graph,795 self._debug_graphs[debug_graph.device_name] = debug_graph796 self._collect_node_devices(debug_graph)798 if validate and debug_graph.device_name in self._dump_tensor_data:799 self._validate_dump_with_graphs(debug_graph.device_name)811 def _collect_node_devices(self, debug_graph): argument812 for node_name in debug_graph.node_devices:815 debug_graph.node_devices[node_name])817 self._node_devices[node_name] = debug_graph.node_devices[node_name]837 debug_graph = self._debug_graphs[device_name][all …]