diff options
| author |  Guangda Lai <31743510+aaroey@users.noreply.github.com> | 2018-09-19 14:39:49 -0700 |
|---|---|---|
| committer | Guangda Lai <31743510+aaroey@users.noreply.github.com> | 2018-09-19 14:39:49 -0700 |
| commit | 5f05a18c576ba89a7bce5f2ed5c7104bc158d8f1 (patch) | |
| tree | 8f775ff7edcec97965f4e137cee377bd17d63124 /tensorflow/contrib/tensorrt | |
| parent | 1593b4a1508a57694c3263d045d55334cab8cfcd (diff) | |
Fix variable name.
Diffstat (limited to 'tensorflow/contrib/tensorrt')
| -rw-r--r-- | tensorflow/contrib/tensorrt/convert/convert_nodes.cc | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/tensorrt/convert/convert_nodes.cc b/tensorflow/contrib/tensorrt/convert/convert_nodes.cc index 21fb459483..6283bd2300 100644 --- a/tensorflow/contrib/tensorrt/convert/convert_nodes.cc +++ b/tensorflow/contrib/tensorrt/convert/convert_nodes.cc @@ -780,11 +780,11 @@ class Converter { // skip control nodes if (input_name[0] == '^') continue; string name = input_name; - auto first = name.find_last_of(':'); + auto last = name.find_last_of(':'); // TODO(aaroey): use TensorId - if (first != string::npos && first + 2 == name.size() && - name[first + 1] == '0') { - name.erase(first); + if (last != string::npos && last + 2 == name.size() && + name[last + 1] == '0') { + name.erase(last); } if (trt_tensors_.count(name)) { |