52 #include <dw/core/Config.h> 56 #include <driver_types.h> 154 uint32_t modelContentSize,
205 dwDNNHandle_t network);
231 dwDNNHandle_t network);
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 - sin...
DW_API_PUBLIC dwStatus dwDNN_getOutputBlobCount(uint32_t *count, dwDNNHandle_t network)
Gets the output blob count.
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.
float float32_t
Specifies POD types.
Specified plugin configuration.
DW_API_PUBLIC dwStatus dwDNN_getInputBlobCount(uint32_t *count, dwDNNHandle_t network)
Gets the input blob count.
DW_API_PUBLIC dwStatus dwDNN_getCUDAStream(cudaStream_t *stream, dwDNNHandle_t network)
Gets the CUDA stream used by the feature list.
DW_API_PUBLIC dwStatus dwDNN_getOutputTensorProperties(dwDNNTensorProperties *tensorProps, uint32_t blobIndex, dwDNNHandle_t network)
Gets the output tensor properties at blobIndex.
NVIDIA DriveWorks API: Core Methods
DW_API_PUBLIC dwStatus dwDNN_getInputSize(dwBlobSize *blobSize, uint32_t blobIndex, dwDNNHandle_t network)
Gets the input blob size at blobIndex.
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.
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.
DW_API_PUBLIC dwStatus dwDNN_reset(dwDNNHandle_t network)
Resets a given network.
NVIDIA DriveWorks API: DNNTensor Structures and Methods
DW_API_PUBLIC dwStatus dwDNN_getInputTensorProperties(dwDNNTensorProperties *tensorProps, uint32_t blobIndex, dwDNNHandle_t network)
Gets the input tensor properties at blobIndex.
DW_API_PUBLIC dwStatus dwDNN_setCUDAStream(cudaStream_t stream, dwDNNHandle_t network)
Sets the CUDA stream for infer operations.
const char8_t * pluginLibraryPath
Path to a plugin shared object.
dwStatus
Status definition.
size_t numCustomLayers
Number of custom layers.
dwProcessorType
Processor type definitions.
NVIDIA DriveWorks API: Data Conditioner Methods
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.
struct dwDNNTensorObject const * dwConstDNNTensorHandle_t
DW_API_PUBLIC dwStatus dwDNN_getOutputSize(dwBlobSize *blobSize, uint32_t blobIndex, dwDNNHandle_t network)
Gets the output blob size at blobIndex.
struct dwDNNObject const * dwConstDNNHandle_t
const char8_t * layerName
Name of the custom layer.
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.
Specifies DNNTensor properties.
struct dwDNNObject * dwDNNHandle_t
Handles representing Deep Neural Network interface.
const dwDNNCustomLayer * customLayers
Array of custom layers.
struct dwContextObject * dwContextHandle_t
Context handle.
struct dwDNNTensorObject * dwDNNTensorHandle_t
Handles representing Deep Neural Network interface.
dwDataConditionerParams dataConditionerParams
DataConditioner parameters for running this network.
DW_API_PUBLIC dwStatus dwDNN_getMetaData(dwDNNMetaData *metaData, dwDNNHandle_t network)
Returns the metadata for the associated network model.
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.
DW_API_PUBLIC dwStatus dwDNN_release(dwDNNHandle_t network)
Releases a given network.
NVIDIA DriveWorks API: Core Status Methods