| 312 | } |
| 313 | |
| 314 | void getLayerPrecisions(Arguments& arguments, char const* argument, LayerPrecisions& layerPrecisions) |
| 315 | { |
| 316 | std::string list; |
| 317 | if (!getAndDelOption(arguments, argument, list)) |
| 318 | { |
| 319 | return; |
| 320 | } |
| 321 | |
| 322 | // The layerPrecisions flag contains comma-separated layerName:precision pairs. |
| 323 | std::vector<std::string> precisionList{splitToStringVec(list, ',')}; |
| 324 | for (auto const& s : precisionList) |
| 325 | { |
| 326 | auto namePrecisionPair = splitNameAndValue<nvinfer1::DataType>(s); |
| 327 | auto const layerName = removeSingleQuotationMarks(namePrecisionPair.first); |
| 328 | layerPrecisions[layerName] = namePrecisionPair.second; |
| 329 | } |
| 330 | } |
| 331 | |
| 332 | void getLayerOutputTypes(Arguments& arguments, char const* argument, LayerOutputTypes& layerOutputTypes) |
| 333 | { |
no test coverage detected