diff --git a/tensorflow/core/graph/graph_constructor.cc b/tensorflow/core/graph/graph_constructor.cc index add26f3b711..8c73f8f7125 100644 --- a/tensorflow/core/graph/graph_constructor.cc +++ b/tensorflow/core/graph/graph_constructor.cc @@ -1042,6 +1042,14 @@ Status GraphConstructor::Convert() { } if (processed < node_defs_.size()) { + LOG(WARNING) << "IN " << __func__ << (node_defs_.size() - processed) + << " NODES IN A CYCLE"; + for (int64 i = 0; i < node_defs_.size(); i++) { + if (pending_count_[i] != 0) { + LOG(WARNING) << "PENDING: " << SummarizeNodeDef(*node_defs_[i]) + << "WITH PENDING COUNT = " << pending_count_[i]; + } + } return errors::InvalidArgument(node_defs_.size() - processed, " nodes in a cycle"); } diff --git a/tensorflow/tools/graph_transforms/transform_utils.cc b/tensorflow/tools/graph_transforms/transform_utils.cc index af17fd75bc1..cb084e49b7c 100644 --- a/tensorflow/tools/graph_transforms/transform_utils.cc +++ b/tensorflow/tools/graph_transforms/transform_utils.cc @@ -247,9 +247,16 @@ Status SortByExecutionOrder(const GraphDef& input_graph_def, } } - if (processed < input_graph_def.node_size()) { - return errors::InvalidArgument(input_graph_def.node_size() - processed, - " nodes in a cycle"); + if (processed < num_nodes) { + LOG(WARNING) << "IN " << __func__ << (num_nodes - processed) + << " NODES IN A CYCLE"; + for (int64 i = 0; i < num_nodes; i++) { + if (pending_count[i] != 0) { + LOG(WARNING) << "PENDING: " << SummarizeNodeDef(input_graph_def.node(i)) + << "WITH PENDING COUNT = " << pending_count[i]; + } + } + return errors::InvalidArgument(num_nodes - processed, " nodes in a cycle"); } return Status::OK(); }