NVIDIA DriveWorks: DNN Plugin Interface
Description: This file defines the DNN custom layer plugin interface layer.
Definition in file DNNPlugin.h.
Go to the source code of this file.
Data Structures | |
struct | dwDNNPluginWeights |
Stores DNN weights. More... | |
Typedefs | |
typedef void * | _dwDNNPluginHandle_t |
_dwDNNPluginHandle_t can be optionally used for storing and accessing variables among the functions defined below. More... | |
Enumerations | |
enum | dwDNNPluginLayout { DW_DNN_PLUGIN_LAYOUT_LINEAR = 0, DW_DNN_PLUGIN_LAYOUT_NCHW = DW_DNN_PLUGIN_LAYOUT_LINEAR, DW_DNN_PLUGIN_LAYOUT_NC2HW2 = 1, DW_DNN_PLUGIN_LAYOUT_CHW2 = DW_DNN_PLUGIN_LAYOUT_NC2HW2, DW_DNN_PLUGIN_LAYOUT_NHWC8 = 2, DW_DNN_PLUGIN_LAYOUT_HWC8 = DW_DNN_PLUGIN_LAYOUT_NHWC8, DW_DNN_PLUGIN_LAYOUT_CHW4 = 3, DW_DNN_PLUGIN_LAYOUT_CHW16 = 4, DW_DNN_PLUGIN_LAYOUT_CHW32 = 5 } |
Represents different layouts for plugins. More... | |
Functions | |
dwStatus | _dwDNNPlugin_configureWithFormat (const dwBlobSize *inputDimensions, int32_t numInputs, const dwBlobSize *outputDimensions, int32_t numOutputs, dwPrecision precision, dwDNNPluginLayout layout, int32_t maxBatchSize, _dwDNNPluginHandle_t handle) |
Configures the plugin with given format. More... | |
dwStatus | _dwDNNPlugin_enqueue (int32_t batchSize, const void *const *inputs, void **outputs, void *workspace, cudaStream_t stream, _dwDNNPluginHandle_t handle) |
Performs forward-pass. More... | |
dwStatus | _dwDNNPlugin_getNumOutputs (int32_t *numOutputs, _dwDNNPluginHandle_t handle) |
Returns number of outputs. More... | |
dwStatus | _dwDNNPlugin_getOutputDimensions (dwBlobSize *outputDimensions, int32_t outputIndex, const dwBlobSize *inputDimensions, int32_t numInputs, _dwDNNPluginHandle_t handle) |
Returns output dimensions of an output at a given index based on inputDimensions. More... | |
dwStatus | _dwDNNPlugin_getSerializationSize (size_t *serializationSize, _dwDNNPluginHandle_t handle) |
Returns serialization size. More... | |
dwStatus | _dwDNNPlugin_getWorkspaceSize (size_t *workspaceSize, int32_t maxBatchSize, _dwDNNPluginHandle_t handle) |
Returns workspace size. More... | |
dwStatus | _dwDNNPlugin_initialize (_dwDNNPluginHandle_t *handle, const char8_t *layerName, const void *data, size_t length) |
Initializes the custom plugin from serialized bytes. More... | |
dwStatus | _dwDNNPlugin_initializeFromWeights (_dwDNNPluginHandle_t *handle, const char8_t *layerName, const dwDNNPluginWeights *weights, int32_t numWeights) |
Initializes the custom plugin from weights. More... | |
dwStatus | _dwDNNPlugin_release (_dwDNNPluginHandle_t handle) |
Releases the custom plugin. More... | |
dwStatus | _dwDNNPlugin_serialize (void *buffer, _dwDNNPluginHandle_t handle) |
Serializes the layer. More... | |
dwStatus | _dwDNNPlugin_supportsFormat (bool *res, dwPrecision precision, dwDNNPluginLayout pluginLayout, _dwDNNPluginHandle_t handle) |
Returns a flag indicating whether the given format is supported. More... | |