| 400 | } |
| 401 | |
| 402 | void fillShapes(BuildOptions::ShapeProfile& shapes, std::string const& name, ShapeRange const& sourceShapeRange, |
| 403 | nvinfer1::OptProfileSelector minDimsSource, nvinfer1::OptProfileSelector optDimsSource, |
| 404 | nvinfer1::OptProfileSelector maxDimsSource) |
| 405 | { |
| 406 | insertShapesBuild( |
| 407 | shapes, nvinfer1::OptProfileSelector::kMIN, name, sourceShapeRange[static_cast<size_t>(minDimsSource)]); |
| 408 | insertShapesBuild( |
| 409 | shapes, nvinfer1::OptProfileSelector::kOPT, name, sourceShapeRange[static_cast<size_t>(optDimsSource)]); |
| 410 | insertShapesBuild( |
| 411 | shapes, nvinfer1::OptProfileSelector::kMAX, name, sourceShapeRange[static_cast<size_t>(maxDimsSource)]); |
| 412 | } |
| 413 | |
| 414 | void processShapes(BuildOptions::ShapeProfile& shapes, bool minShapes, bool optShapes, bool maxShapes, bool calib) |
| 415 | { |
no test coverage detected