fix(tensorrt_common): fix shadowVariable (#7906)

* fix:shadowVariable Signed-off-by: kobayu858 <yutaro.kobayashi@tier4.jp> * fix:shadowVariable Signed-off-by: kobayu858 <yutaro.kobayashi@tier4.jp> --------- Signed-off-by: kobayu858 <yutaro.kobayashi@tier4.jp>
autowarefoundation · Jul 10, 2024 · b1d7fcd · b1d7fcd
1 parent defce44
commit b1d7fcd
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/common/tensorrt_common/src/tensorrt_common.cpp b/common/tensorrt_common/src/tensorrt_common.cpp
@@ -427,18 +427,18 @@ bool TrtCommon::buildEngineFromOnnx(
             layer->setPrecision(nvinfer1::DataType::kHALF);
             std::cout << "Set kHALF in " << name << std::endl;
           }
-          for (int i = num - 1; i >= 0; i--) {
-            nvinfer1::ILayer * layer = network->getLayer(i);
-            auto layer_type = layer->getType();
-            std::string name = layer->getName();
-            if (layer_type == nvinfer1::LayerType::kCONVOLUTION) {
-              layer->setPrecision(nvinfer1::DataType::kHALF);
-              std::cout << "Set kHALF in " << name << std::endl;
+          for (int j = num - 1; j >= 0; j--) {
+            nvinfer1::ILayer * inner_layer = network->getLayer(j);
+            auto inner_layer_type = inner_layer->getType();
+            std::string inner_name = inner_layer->getName();
+            if (inner_layer_type == nvinfer1::LayerType::kCONVOLUTION) {
+              inner_layer->setPrecision(nvinfer1::DataType::kHALF);
+              std::cout << "Set kHALF in " << inner_name << std::endl;
               break;
             }
-            if (layer_type == nvinfer1::LayerType::kMATRIX_MULTIPLY) {
-              layer->setPrecision(nvinfer1::DataType::kHALF);
-              std::cout << "Set kHALF in " << name << std::endl;
+            if (inner_layer_type == nvinfer1::LayerType::kMATRIX_MULTIPLY) {
+              inner_layer->setPrecision(nvinfer1::DataType::kHALF);
+              std::cout << "Set kHALF in " << inner_name << std::endl;
               break;
             }
           }