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

Function fillShapes

samples/common/sampleOptions.cpp:402–412  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

400}
401
402void 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
414void processShapes(BuildOptions::ShapeProfile& shapes, bool minShapes, bool optShapes, bool maxShapes, bool calib)
415{

Callers 1

processShapesFunction · 0.85

Calls 1

insertShapesBuildFunction · 0.85

Tested by

no test coverage detected