fix:shadowVariable

Signed-off-by: kobayu858 <yutaro.kobayashi@tier4.jp>
autowarefoundation · Jul 10, 2024 · 715f875 · 715f875
1 parent 473b0a7
commit 715f875
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/common/tensorrt_common/src/tensorrt_common.cpp b/common/tensorrt_common/src/tensorrt_common.cpp
@@ -432,12 +432,12 @@ bool TrtCommon::buildEngineFromOnnx(
             auto inner_layer_type = inner_layer->getType();
             std::string inner_name = inner_layer->getName();
             if (inner_layer_type == nvinfer1::LayerType::kCONVOLUTION) {
-              layer->setPrecision(nvinfer1::DataType::kHALF);
+              inner_layer->setPrecision(nvinfer1::DataType::kHALF);
               std::cout << "Set kHALF in " << inner_name << std::endl;
               break;
             }
             if (inner_layer_type == nvinfer1::LayerType::kMATRIX_MULTIPLY) {
-              layer->setPrecision(nvinfer1::DataType::kHALF);
+              inner_layer->setPrecision(nvinfer1::DataType::kHALF);
               std::cout << "Set kHALF in " << inner_name << std::endl;
               break;
             }