TensorStreamer.h File Reference

Detailed Description

NVIDIA DriveWorks API: Tensor Streamer

Description: This file defines the tensor streamer function.

Definition in file TensorStreamer.h.

Go to the source code of this file.

Typedefs
typedef struct dwDNNTensorStreamerObject *	dwDNNTensorStreamerHandle_t

Functions
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...