Method getTensorLocation

include/NvInferRuntime.h:1805–1808

\brief Get whether an input or output tensor must be on GPU or CPU. \param tensorName The name of an input or output tensor. \return TensorLocation::kDEVICE if tensorName must be on GPU, or TensorLocation::kHOST if on CPU, or TensorLocation::kDEVICE if the provided name does not map to an input or output tensor. The location is established at build time. E.g. shape tensors inputs are typically

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

    //!
    //! \brief True if tensor is required as input for shape calculations or is output from shape calculations.