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

Method inferShapes

include/NvInferRuntime.h:3336–3339  ·  view source on GitHub ↗

\brief Run shape calculations. \param nbMaxNames Maximum number of names to write to tensorNames. When the return value is a positive value n and tensorNames != nullptr, the names of min(n,nbMaxNames) insufficiently specified input tensors are written to tensorNames. \param tensorNames Buffer in which to place names of insufficiently specified input tensors. \return 0 on success. Positive value

Source from the content-addressed store, hash-verified

3334 //! Returns -1 if a tensor's dimensions are invalid, e.g. a tensor ends up with a negative dimension.
3335 //!
3336 int32_t inferShapes(int32_t nbMaxNames, char const** tensorNames) noexcept
3337 {
3338 return mImpl->inferShapes(nbMaxNames, tensorNames);
3339 }
3340
3341 //!
3342 //! \brief Mark input as consumed.

Callers 1

infer_shapesFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected