diff options
author | 2018-07-16 12:44:41 -0700 | |
---|---|---|
committer | 2018-07-16 12:44:41 -0700 | |
commit | 24764abb514c2664d8219700f9da7a6c3a8c6146 (patch) | |
tree | 3c099f2d6558e4ffd344cbf064613ae3d7cd747c /tensorflow/contrib/tensorrt | |
parent | 7c505ddfe6cf32f82a1ea27f61c5f4fe93409098 (diff) |
Clang format, updated comment
Diffstat (limited to 'tensorflow/contrib/tensorrt')
-rw-r--r-- | tensorflow/contrib/tensorrt/convert/trt_optimization_pass.cc | 16 |
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); } |