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

Function validateInputParams

samples/sampleINT8API/sampleINT8API.cpp:754–772  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

752}
753
754void validateInputParams(SampleINT8APIParams& params)
755{
756 sample::gLogInfo << "Please follow README.md to generate missing input files." << std::endl;
757 sample::gLogInfo << "Validating input parameters. Using following input files for inference." << std::endl;
758 params.modelFileName = locateFile(params.modelFileName, params.dataDirs);
759 sample::gLogInfo << " Model File: " << params.modelFileName << std::endl;
760 if (params.writeNetworkTensors)
761 {
762 sample::gLogInfo << " Writing Network Tensors File to: " << params.networkTensorsFileName << std::endl;
763 return;
764 }
765 params.imageFileName = locateFile(params.imageFileName, params.dataDirs);
766 sample::gLogInfo << " Image File: " << params.imageFileName << std::endl;
767 params.referenceFileName = locateFile(params.referenceFileName, params.dataDirs);
768 sample::gLogInfo << " Reference File: " << params.referenceFileName << std::endl;
769 params.dynamicRangeFileName = locateFile(params.dynamicRangeFileName, params.dataDirs);
770 sample::gLogInfo << " Dynamic Range File: " << params.dynamicRangeFileName << std::endl;
771 return;
772}
773
774//!
775//! \brief This function initializes members of the params struct using the command line args

Callers 1

initializeSampleParamsFunction · 0.85

Calls 1

locateFileFunction · 0.85

Tested by

no test coverage detected