DNN.h File Reference

Detailed Description

NVIDIA DriveWorks API: DNN Methods

Description: This file defines Deep Neural Network methods.

Definition in file DNN.h.

Go to the source code of this file.

Data Structures
struct	dwDNNMetaData
	Specifies TensorRT model header. More...

Typedefs
typedef struct dwDNNObject const *	dwConstDNNHandle_t
typedef struct dwDNNObject *	dwDNNHandle_t
	Handles representing Deep Neural Network interface. More...

Functions
DW_API_PUBLIC dwStatus	dwDNN_getCUDAStream (cudaStream_t *stream, dwDNNHandle_t network)
	Gets the CUDA stream used by the feature list. More...
DW_API_PUBLIC dwStatus	dwDNN_getInputBlobCount (uint16_t *count, dwDNNHandle_t network)
	Gets the input blob count. More...
DW_API_PUBLIC dwStatus	dwDNN_getInputIndex (uint16_t *blobIndex, const char *blobName, dwDNNHandle_t network)
	Gets the index of an input blob with a given blob name. More...
DW_API_PUBLIC dwStatus	dwDNN_getInputSize (dwBlobSize *blobSize, uint16_t blobIndex, dwDNNHandle_t network)
	Gets the input blob size at blobIndex. More...
DW_API_PUBLIC dwStatus	dwDNN_getMetaData (dwDNNMetaData *metaData, dwDNNHandle_t network)
	Returns the metadata for the associated network model. More...
DW_API_PUBLIC dwStatus	dwDNN_getOutputBlobCount (uint16_t *count, dwDNNHandle_t network)
	Gets the output blob count. More...
DW_API_PUBLIC dwStatus	dwDNN_getOutputIndex (uint16_t *blobIndex, const char *blobName, dwDNNHandle_t network)
	Gets the index of an output blob with a given blob name. More...
DW_API_PUBLIC dwStatus	dwDNN_getOutputSize (dwBlobSize *blobSize, uint16_t blobIndex, dwDNNHandle_t network)
	Gets the output blob size at blobIndex. More...
DW_API_PUBLIC dwStatus	dwDNN_infer (float32_t **d_output, float32_t **d_input, dwDNNHandle_t network)
	Forwards pass from all input blobs to all output blobs. More...
DW_API_PUBLIC dwStatus	dwDNN_inferSIO (float32_t *d_output, float32_t *d_input, dwDNNHandle_t network)
	Forwards pass from the first input blob to the first output blob (a shortcut for a single input - single output network). More...
DW_API_PUBLIC dwStatus	dwDNN_initializeTensorRTFromFile (dwDNNHandle_t *network, dwContextHandle_t context, const char *modelFilename)
	Creates and initializes a TensorRT Network from file. More...
DW_API_PUBLIC dwStatus	dwDNN_initializeTensorRTFromMemory (dwDNNHandle_t *network, dwContextHandle_t context, const char *modelContent, uint32_t modelContentSize)
	Creates and initializes a TensorRT Network from memory. More...
DW_API_PUBLIC dwStatus	dwDNN_release (dwDNNHandle_t *network)
	Releases a given network. More...
DW_API_PUBLIC dwStatus	dwDNN_reset (dwDNNHandle_t network)
	Resets a given network. More...
DW_API_PUBLIC dwStatus	dwDNN_setCUDAStream (cudaStream_t stream, dwDNNHandle_t network)
	Sets the CUDA stream for infer operations. More...