DNNPluginConfig.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2018-2024 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
 *
 * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
 * property and proprietary rights in and to this material, related
 * documentation and any modifications thereto. Any use, reproduction,
 * disclosure or distribution of this material and related documentation
 * without an express license agreement from NVIDIA CORPORATION or
 * its affiliates is strictly prohibited.
 */
 
#ifndef DW_DNN_PLUGIN_CONFIG_H_
#define DW_DNN_PLUGIN_CONFIG_H_
 
#include <dw/core/base/Types.h>
 
typedef struct
{
    const char8_t* pluginLibraryPath; 
    const char8_t* layerName;         
} dwDNNCustomLayer;
 
typedef struct
{
    const dwDNNCustomLayer* customLayers;    
    size_t numCustomLayers;                  
    const dwDNNCustomLayer* customLayersNew; 
    size_t numCustomLayersNew;               
} dwDNNPluginConfiguration;
 
#endif // DW_DNN_PLUGIN_CONFIG_H_