| 1344 | } |
| 1345 | |
| 1346 | void dumpRefittable(nvinfer1::ICudaEngine& engine) |
| 1347 | { |
| 1348 | std::unique_ptr<IRefitter> refitter{createRefitter(engine)}; |
| 1349 | if (refitter == nullptr) |
| 1350 | { |
| 1351 | sample::gLogError << "Failed to create a refitter." << std::endl; |
| 1352 | return; |
| 1353 | } |
| 1354 | |
| 1355 | auto const& layerWeightsRolePair = getLayerWeightsRolePair(*refitter); |
| 1356 | auto const& layerNames = layerWeightsRolePair.first; |
| 1357 | auto const& weightsRoles = layerWeightsRolePair.second; |
| 1358 | auto const nbAll = layerWeightsRolePair.first.size(); |
| 1359 | for (size_t i = 0; i < nbAll; ++i) |
| 1360 | { |
| 1361 | sample::gLogInfo << layerNames[i] << " " << weightsRoles[i] << std::endl; |
| 1362 | } |
| 1363 | } |
| 1364 | |
| 1365 | ICudaEngine* loadEngine(std::string const& engine, int32_t DLACore, std::ostream& err) |
| 1366 | { |
no test coverage detected