Commit 473b0a7 1 parent 7dcdd67 commit 473b0a7 Copy full SHA for 473b0a7
File tree 1 file changed +8
-8
lines changed
common/tensorrt_common/src
1 file changed +8
-8
lines changed Original file line number Diff line number Diff line change @@ -427,18 +427,18 @@ bool TrtCommon::buildEngineFromOnnx(
427
427
layer->setPrecision (nvinfer1::DataType::kHALF );
428
428
std::cout << " Set kHALF in " << name << std::endl;
429
429
}
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 ) {
435
435
layer->setPrecision (nvinfer1::DataType::kHALF );
436
- std::cout << " Set kHALF in " << name << std::endl;
436
+ std::cout << " Set kHALF in " << inner_name << std::endl;
437
437
break ;
438
438
}
439
- if (layer_type == nvinfer1::LayerType::kMATRIX_MULTIPLY ) {
439
+ if (inner_layer_type == nvinfer1::LayerType::kMATRIX_MULTIPLY ) {
440
440
layer->setPrecision (nvinfer1::DataType::kHALF );
441
- std::cout << " Set kHALF in " << name << std::endl;
441
+ std::cout << " Set kHALF in " << inner_name << std::endl;
442
442
break ;
443
443
}
444
444
}
You can’t perform that action at this time.
0 commit comments