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

Method getTensorIOMode

include/NvInferRuntime.h:1839–1842  ·  view source on GitHub ↗

\brief Determine whether a tensor is an input or output tensor. \param tensorName The name of an input or output tensor. \return kINPUT if tensorName is an input, kOUTPUT if tensorName is an output, or kNONE if neither. \warning The string tensorName must be null-terminated, and be at most 4096 bytes including the terminator.

Source from the content-addressed store, hash-verified

1837 //! \warning The string tensorName must be null-terminated, and be at most 4096 bytes including the terminator.
1838 //!
1839 TensorIOMode getTensorIOMode(char const* tensorName) const noexcept
1840 {
1841 return mImpl->getTensorIOMode(tensorName);
1842 }
1843
1844 //! \brief create an execution context without any device memory allocated
1845 //!

Callers 3

getTensorInfoMethod · 0.80
setUpInferenceFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected