diff options
author | 2018-07-13 22:35:47 -0700 | |
---|---|---|
committer | 2018-07-13 22:35:47 -0700 | |
commit | 511ce2e7eb8f220c443f09382d09d14f0758e8ba (patch) | |
tree | f87377a531297e1eefbde5e33e6cb83e4d1f061c /tensorflow/contrib/tensorrt/convert | |
parent | 7ff7013598717a4c21034fdfb30462442a0888f8 (diff) |
Remove unnecessary for loop
Diffstat (limited to 'tensorflow/contrib/tensorrt/convert')
-rw-r--r-- | tensorflow/contrib/tensorrt/convert/trt_optimization_pass.cc | 6 |
1 files changed, 1 insertions, 5 deletions
diff --git a/tensorflow/contrib/tensorrt/convert/trt_optimization_pass.cc b/tensorflow/contrib/tensorrt/convert/trt_optimization_pass.cc index 84eb8aebe9..5bb0ffc797 100644 --- a/tensorflow/contrib/tensorrt/convert/trt_optimization_pass.cc +++ b/tensorflow/contrib/tensorrt/convert/trt_optimization_pass.cc @@ -236,11 +236,7 @@ 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 (size_t t = 1; t < tokens.size() - 1; ++t) { - StrAppend(&s, ":", tokens.at(t)); - } - nodes_to_preserve.push_back(s); + nodes_to_preserve.push_back(tokens.at(0)); } cp.input_graph_def = &item.graph; cp.output_names = &nodes_to_preserve; |