Skip to content

Commit 473b0a7

Browse files
committed
fix:shadowVariable
Signed-off-by: kobayu858 <yutaro.kobayashi@tier4.jp>
1 parent 7dcdd67 commit 473b0a7

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

common/tensorrt_common/src/tensorrt_common.cpp

+8-8
Original file line numberDiff line numberDiff line change
@@ -427,18 +427,18 @@ bool TrtCommon::buildEngineFromOnnx(
427427
layer->setPrecision(nvinfer1::DataType::kHALF);
428428
std::cout << "Set kHALF in " << name << std::endl;
429429
}
430-
for (int i = num - 1; i >= 0; i--) {
431-
nvinfer1::ILayer * layer = network->getLayer(i);
432-
auto layer_type = layer->getType();
433-
std::string name = layer->getName();
434-
if (layer_type == nvinfer1::LayerType::kCONVOLUTION) {
430+
for (int j = num - 1; j >= 0; j--) {
431+
nvinfer1::ILayer * inner_layer = network->getLayer(j);
432+
auto inner_layer_type = inner_layer->getType();
433+
std::string inner_name = inner_layer->getName();
434+
if (inner_layer_type == nvinfer1::LayerType::kCONVOLUTION) {
435435
layer->setPrecision(nvinfer1::DataType::kHALF);
436-
std::cout << "Set kHALF in " << name << std::endl;
436+
std::cout << "Set kHALF in " << inner_name << std::endl;
437437
break;
438438
}
439-
if (layer_type == nvinfer1::LayerType::kMATRIX_MULTIPLY) {
439+
if (inner_layer_type == nvinfer1::LayerType::kMATRIX_MULTIPLY) {
440440
layer->setPrecision(nvinfer1::DataType::kHALF);
441-
std::cout << "Set kHALF in " << name << std::endl;
441+
std::cout << "Set kHALF in " << inner_name << std::endl;
442442
break;
443443
}
444444
}

0 commit comments

Comments
 (0)