Ignore control out edges from const node.
When we loop throug the input edges, only non-const input edges are
added to the list of connections. With this fix we do the same for
the output edges, otherwise the EngineInfo structures could be
inconsistent if we have multiple TRT segments.
diff --git a/tensorflow/compiler/tf2tensorrt/convert/convert_graph.cc b/tensorflow/compiler/tf2tensorrt/convert/convert_graph.cc
index 6f27654..cb11ab5 100644
--- a/tensorflow/compiler/tf2tensorrt/convert/convert_graph.cc
+++ b/tensorflow/compiler/tf2tensorrt/convert/convert_graph.cc
@@ -221,9 +221,11 @@
}
if (edge->IsControlEdge()) {
// Control output.
- info->connections.emplace_back(output_node->name(), output_node->id(),
- node_name, node_id,
- /*input_edge=*/false);
+ if (node->type_string() != "Const") {
+ info->connections.emplace_back(output_node->name(), output_node->id(),
+ node_name, node_id,
+ /*input_edge=*/false);
+ }
} else {
// Data output.
int port = Graph::kControlSlot - 1;
