aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/tensorrt/convert
diff options
context:
space:
mode:
authorGravatar gracehoney <31743510+aaroey@users.noreply.github.com>2018-07-30 13:31:05 -0700
committerGravatar gracehoney <31743510+aaroey@users.noreply.github.com>2018-07-30 13:31:05 -0700
commit027068aa2ae80b69657fdcf3401249742b1f3d8d (patch)
treedfb18ddce92e7a63e14afa0703db26e3b2afdf55 /tensorflow/contrib/tensorrt/convert
parente8e2cc72f3367aee1789dc0f5bcbd8f027c7180f (diff)
Fix comment.
Diffstat (limited to 'tensorflow/contrib/tensorrt/convert')
-rw-r--r--tensorflow/contrib/tensorrt/convert/convert_graph.cc12
1 files changed, 6 insertions, 6 deletions
diff --git a/tensorflow/contrib/tensorrt/convert/convert_graph.cc b/tensorflow/contrib/tensorrt/convert/convert_graph.cc
index e06704f5d1..e9dc505642 100644
--- a/tensorflow/contrib/tensorrt/convert/convert_graph.cc
+++ b/tensorflow/contrib/tensorrt/convert/convert_graph.cc
@@ -317,18 +317,18 @@ tensorflow::Status GetEngineInfo(
// other engines or TF nodes. Since we add it only to the segment
// graphdef, not the segment itself, it won't be removed from the graph.
// If it doesn't have any edges, TF will prune it out.
- // Note that the constant data input must be supported by the engine
- // regardless of the datatype, since the segmenter already removed
- // unsupported data input nodes.
+ //
+ // Note that the segmenter already ensure that the constant data input
+ // is valid and suppported by the engine.
if (!added_const_node_ids.insert(input_node->id()).second) {
// Already added before.
continue;
}
VLOG(1) << "Adding const node " << input_node->name();
QCHECK(subgraph_node_names.insert(input_node->name()).second);
- // Since we duplicate the const input node in both the segment graphdef
- // and the engine, the segment node doesn't depend on it anymore, so we
- // add a control dependency instead.
+ // Since we already add (duplicate) the const input node to the segment
+ // graphdef, it's now not a data dependency any more, but to make the
+ // dependency correct we still add a control dependency.
info->connections.emplace_back(
input_node->name(), input_node->id(), node_name, node_id,
/*input_edge=*/true);