\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
| 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 |