Method setDeviceType

include/NvInfer.h:9302–9305

\brief Set the device that this layer must execute on. \param layer which layer to execute. \param deviceType that this layer must execute on. If DeviceType is not set or is reset, TensorRT will use the default DeviceType set in the builder. \note The device type for a layer must be compatible with the safety flow (if specified). For example a layer cannot be marked for DLA execution while the bu

    //! \see getDeviceType()
    //!
    void setDeviceType(ILayer const* layer, DeviceType deviceType) noexcept
    {
        mImpl->setDeviceType(layer, deviceType);
    }

    //!
    //! \brief Get the device that this layer executes on.