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 custom layer. 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... | |