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

Method setShapeValues

include/NvInferRuntime.h:1380–1384  ·  view source on GitHub ↗

\brief Set the minimum / optimum / maximum values for an input shape tensor. This function must be called three times for every input tensor t that is a shape tensor (t.isShape() == true). This implies that the datatype of t is DataType::kINT32, the rank is either 0 or 1, and the dimensions of t are fixed at network definition time. This function must not be called for any input tensor that is no

Source from the content-addressed store, hash-verified

1378 //! \warning The string inputName must be null-terminated, and be at most 4096 bytes including the terminator.
1379 //!
1380 bool setShapeValues(
1381 char const* inputName, OptProfileSelector select, int32_t const* values, int32_t nbValues) noexcept
1382 {
1383 return mImpl->setShapeValues(inputName, select, values, nbValues);
1384 }
1385
1386 //!
1387 //! \brief Get the number of values for an input shape tensor.

Callers 2

pyCore.cppFile · 0.80
setupNetworkAndConfigFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected