NVIDIA DriveWorks API: Tensor Streamer
Description: This file defines the tensor streamer function.
Definition in file TensorStreamer.h.
|
| DW_API_PUBLIC dwStatus | dwDNNTensorStreamer_consumerReceive (dwDNNTensorHandle_t *tensor, dwTime_t timeout_us, dwDNNTensorStreamerHandle_t streamer) |
| | Receive a pointer to a dwDNNTensorHandle_t from the streamer, acting as a consumer. More...
|
| |
| DW_API_PUBLIC dwStatus | dwDNNTensorStreamer_consumerReturn (dwDNNTensorHandle_t *tensor, dwDNNTensorStreamerHandle_t streamer) |
| | Return the received tensor back to the producer. More...
|
| |
| DW_API_PUBLIC dwStatus | dwDNNTensorStreamer_getCUDAStream (cudaStream_t *stream, dwDNNTensorStreamerHandle_t streamer) |
| | Get CUDA stream used by the tensor streamer. More...
|
| |
| DW_API_PUBLIC dwStatus | dwDNNTensorStreamer_getOutputProperties (dwDNNTensorProperties *props, dwDNNTensorStreamerHandle_t const streamer) |
| | Get tensor properties of the tensor received from the streamer. More...
|
| |
| DW_API_PUBLIC dwStatus | dwDNNTensorStreamer_initialize (dwDNNTensorStreamerHandle_t *streamer, const dwDNNTensorProperties *from, dwDNNTensorType to, dwContextHandle_t ctx) |
| | Creates and initializes the tensor streamer capable of moving tensors between different API types. More...
|
| |
| DW_API_PUBLIC dwStatus | dwDNNTensorStreamer_producerReturn (dwDNNTensorHandle_t *tensor, dwTime_t timeout_us, dwDNNTensorStreamerHandle_t streamer) |
| | The producer streamer waits for the tensor sent to be returned by the consumer. More...
|
| |
| DW_API_PUBLIC dwStatus | dwDNNTensorStreamer_producerSend (dwDNNTensorHandle_t tensor, dwDNNTensorStreamerHandle_t streamer) |
| | Sends an tensor through the streamer acting as the producer. More...
|
| |
| DW_API_PUBLIC dwStatus | dwDNNTensorStreamer_release (dwDNNTensorStreamerHandle_t streamer) |
| | Releases the tensor streamer. More...
|
| |
| DW_API_PUBLIC dwStatus | dwDNNTensorStreamer_setCUDAStream (cudaStream_t stream, dwDNNTensorStreamerHandle_t streamer) |
| | Sets the CUDA stream for CUDA related streaming operations such as post and receive. More...
|
| |