Clang format, updated comment

1 files changed, 9 insertions, 7 deletions

diff --git a/tensorflow/contrib/tensorrt/convert/trt_optimization_pass.cc b/tensorflow/contrib/tensorrt/convert/trt_optimization_pass.cc
index 4957daae10..b39c8bdd64 100644
--- a/tensorflow/contrib/tensorrt/convert/trt_optimization_pass.cc
+++ b/tensorflow/contrib/tensorrt/convert/trt_optimization_pass.cc
@@ -237,14 +237,16 @@ tensorflow::Status TRTOptimizationPass::Optimize(
   std::vector<string> nodes_to_preserve;
   for (const auto& n : item.NodesToPreserve()) {
     auto tokens = str_util::Split(n, ":");
-    string s=tokens.at(0);
-    for(int i=1;i<tokens.size()-1;++i){
-      StrAppend(&s,":",tokens.at(i));
+    string s = tokens.at(0);
+    for (int i = 1; i < tokens.size() - 1; ++i){
+      StrAppend(&s, ":", tokens.at(i));
     }
-    int dumm_port=-1;
-    // make sure last token is a port and append it to string if not
-    if(tokens.size() > 1 && !strings::safe_strto32(tokens.back(),&dumm_port)){
-      StrAppend(&s,":",tokens.back());
+    int dumm_port = -1;
+    // If the last token is not an integer, it must be part of the name.
+    // Otherwise it is port number.
+    if (tokens.size() > 1 &&
+       !strings::safe_strto32(tokens.back(), &dumm_port)) {
+      StrAppend(&s, ":", tokens.back());
     }
     nodes_to_preserve.push_back(s);
   }