Method getTensorDataType

include/NvInferRuntime.h:1697–1700

\brief Determine the required data type for a buffer from its tensor name. \param tensorName The name of an input or output tensor. \return The type of the data in the buffer, or DataType::kFLOAT if the provided name does not map to an input or output tensor. \warning The string tensorName must be null-terminated, and be at most 4096 bytes including the terminator.

    //! \warning The string tensorName must be null-terminated, and be at most 4096 bytes including the terminator.
    //!
    DataType getTensorDataType(char const* tensorName) const noexcept
    {
        return mImpl->getTensorDataType(tensorName);
    }

    //!
    //! \brief Get the maximum batch size which can be used for inference. Should only be called if the engine is built