MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / getLayerInformation

Function getLayerInformation

samples/common/sampleInference.cpp:1272–1283  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1270}
1271
1272std::string getLayerInformation(
1273 nvinfer1::ICudaEngine* engine, nvinfer1::IExecutionContext* context, nvinfer1::LayerInformationFormat format)
1274{
1275 auto runtime = std::unique_ptr<IRuntime>{createRuntime()};
1276 auto inspector = std::unique_ptr<IEngineInspector>(engine->createEngineInspector());
1277 if (context != nullptr)
1278 {
1279 inspector->setExecutionContext(context);
1280 }
1281 std::string result = inspector->getEngineInformation(format);
1282 return result;
1283}
1284
1285void Binding::fill(std::string const& fileName)
1286{

Callers 1

printLayerInfoFunction · 0.85

Calls 3

setExecutionContextMethod · 0.80
getEngineInformationMethod · 0.80
createRuntimeFunction · 0.70

Tested by

no test coverage detected