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 | dwDNNCustomLayer |
| Specifies plugin configuration. More... | |
| struct | dwDNNMetaData |
| Specifies TensorRT model header. More... | |
| struct | dwDNNPluginConfiguration |
| Specified plugin configuration. 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 (uint32_t *count, dwDNNHandle_t network) |
| Gets the input blob count. More... | |
| DW_API_PUBLIC dwStatus | dwDNN_getInputIndex (uint32_t *blobIndex, const char8_t *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, uint32_t blobIndex, dwDNNHandle_t network) |
| Gets the input blob size at blobIndex. More... | |
| DW_API_PUBLIC dwStatus | dwDNN_getInputTensorProperties (dwDNNTensorProperties *tensorProps, uint32_t blobIndex, dwDNNHandle_t network) |
| Gets the input tensor properties 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 (uint32_t *count, dwDNNHandle_t network) |
| Gets the output blob count. More... | |
| DW_API_PUBLIC dwStatus | dwDNN_getOutputIndex (uint32_t *blobIndex, const char8_t *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, uint32_t blobIndex, dwDNNHandle_t network) |
| Gets the output blob size at blobIndex. More... | |
| DW_API_PUBLIC dwStatus | dwDNN_getOutputTensorProperties (dwDNNTensorProperties *tensorProps, uint32_t blobIndex, dwDNNHandle_t network) |
| Gets the output tensor properties at blobIndex. More... | |
| DW_API_PUBLIC dwStatus | dwDNN_infer (dwDNNTensorHandle_t *outputTensors, uint32_t outputTensorCount, dwConstDNNTensorHandle_t *inputTensors, uint32_t inputTensorCount, dwDNNHandle_t network) |
| Runs inference pipeline on the given input. More... | |
| DW_API_PUBLIC dwStatus | dwDNN_inferRaw (float32_t *const *d_output, const float32_t *const *d_input, uint32_t batchsize, dwDNNHandle_t network) |
| Forwards pass from all input blobs to all output blobs. More... | |
| DW_API_PUBLIC dwStatus | dwDNN_inferSIO (float32_t *d_output, const float32_t *d_input, uint32_t batchsize, 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, const char8_t *modelFilename, const dwDNNPluginConfiguration *pluginConfiguration, dwProcessorType processorType, dwContextHandle_t context) |
| Creates and initializes a TensorRT Network from file. More... | |
| DW_API_PUBLIC dwStatus | dwDNN_initializeTensorRTFromMemory (dwDNNHandle_t *network, const char8_t *modelContent, uint32_t modelContentSize, const dwDNNPluginConfiguration *pluginConfiguration, dwProcessorType processorType, dwContextHandle_t context) |
| 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... | |