/** ****************************************************************************** * @file layers_upsample_generic.h * @author STMicroelectronics * @brief header file of AI platform padding generic datatypes ****************************************************************************** * @attention * * Copyright (c) 2022 STMicroelectronics. * All rights reserved. * * This software is licensed under terms that can be found in the LICENSE file * in the root directory of this software component. * If no LICENSE file comes with this software, it is provided AS-IS. * ****************************************************************************** */ #ifndef LAYERS_UPSAMPLE_GENERIC_H #define LAYERS_UPSAMPLE_GENERIC_H #include "layers_generic.h" /*! * @defgroup layers_pad_generic Layers Definitions * @brief definition * */ AI_API_DECLARE_BEGIN /******************************************************************************/ /* Forward Functions Section */ /******************************************************************************/ /*! * @brief Handles generic upsampling in nearest mode * @ingroup layers_generic * @param layer upsample layer */ AI_INTERNAL_API void forward_upsample_nearest(ai_layer *pLayer); /*! * @brief Handles generic upsampling in zeros mode * @ingroup layers_generic * @param layer upsample layer */ AI_INTERNAL_API void forward_upsample_zeros(ai_layer *pLayer); /*! * @brief Handles generic upsampling in bilinear mode * @ingroup layers_generic * @param layer upsample layer */ AI_INTERNAL_API void forward_upsample_bilinear(ai_layer *pLayer); AI_API_DECLARE_END #endif /*LAYERS_UPSAMPLE_GENERIC_H*/