aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/tensorrt/convert
diff options
context:
space:
mode:
authorGravatar Sami Kama <skama@nvidia.com>2018-07-13 22:35:47 -0700
committerGravatar Sami Kama <skama@nvidia.com>2018-07-13 22:35:47 -0700
commit511ce2e7eb8f220c443f09382d09d14f0758e8ba (patch)
treef87377a531297e1eefbde5e33e6cb83e4d1f061c /tensorflow/contrib/tensorrt/convert
parent7ff7013598717a4c21034fdfb30462442a0888f8 (diff)
Remove unnecessary for loop
Diffstat (limited to 'tensorflow/contrib/tensorrt/convert')
-rw-r--r--tensorflow/contrib/tensorrt/convert/trt_optimization_pass.cc6
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;