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

Method getProfileShape

include/NvInferRuntime.h:2221–2224  ·  view source on GitHub ↗

\brief Get the minimum / optimum / maximum dimensions for an input tensor given its name under an optimization profile. \param tensorName The name of an input tensor. \param profileIndex The profile index, which must be between 0 and getNbOptimizationProfiles()-1. \param select Whether to query the minimum, optimum, or maximum dimensions for this input tensor. \return The minimum / optimum / m

Source from the content-addressed store, hash-verified

2219 //! \warning The string tensorName must be null-terminated, and be at most 4096 bytes including the terminator.
2220 //!
2221 Dims getProfileShape(char const* tensorName, int32_t profileIndex, OptProfileSelector select) const noexcept
2222 {
2223 return mImpl->getProfileShape(tensorName, profileIndex, select);
2224 }
2225
2226 //!
2227 //! \brief Get minimum / optimum / maximum values for an input shape binding under an optimization profile.

Callers 1

get_tensor_profile_shapeFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected