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

Method getTensorVectorizedDim

include/NvInferRuntime.h:2126–2129  ·  view source on GitHub ↗

\brief Return the dimension index that the buffer is vectorized, or -1 if the provided name does not map to an input or output tensor. Specifically -1 is returned if scalars per vector is 1. \param tensorName The name of an input or output tensor. \warning The string tensorName must be null-terminated, and be at most 4096 bytes including the terminator. \warning The function can only return the

Source from the content-addressed store, hash-verified

2124 //! multiple profiles in the engine, use getTensorVectorizedDim with profileIndex when there are multiple profiles.
2125 //!
2126 int32_t getTensorVectorizedDim(char const* tensorName) const noexcept
2127 {
2128 return mImpl->getTensorVectorizedDim(tensorName);
2129 }
2130
2131 //!
2132 //! \brief Return the dimension index that the buffer is vectorized of given profile, or -1 if the provided name

Callers 2

getNbErrorsFunction · 0.80
getTensorInfoMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected