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

Method setOutputAllocator

include/NvInferRuntime.h:3381–3384  ·  view source on GitHub ↗

\brief Set output allocator to use for output tensor of given name. Pass nullptr to outputAllocator to unset. The allocator is called by enqueueV3(). \param tensorName The name of an output tensor. \param outputAllocator IOutputAllocator for the tensors. \return True if success, false if the provided name does not map to an output or, if some other error occurred. \warning The string tensorName

Source from the content-addressed store, hash-verified

3379 //! \see enqueueV3() IOutputAllocator
3380 //!
3381 bool setOutputAllocator(char const* tensorName, IOutputAllocator* outputAllocator) noexcept
3382 {
3383 return mImpl->setOutputAllocator(tensorName, outputAllocator);
3384 }
3385
3386 //!
3387 //! \brief Get output allocator associated with output tensor of given name, or nullptr if the provided name does

Callers 1

setTensorAddressesMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected