openmv/lib/stai/libstai/include/layers_argminmax.h
iabdalkader e95a19c963 lib: Add STAI library and ML backend.
Signed-off-by: iabdalkader <i.abdalkader@gmail.com>
2025-06-10 11:53:31 +02:00

50 lines
1.5 KiB
C

/**
******************************************************************************
* @file layers_arminmax.h
* @author AIS
* @brief header file of AI platform generic layers datatypes
******************************************************************************
* @attention
*
* Copyright (c) 2023 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_ARGMINMAX_H
#define LAYERS_ARGMINMAX_H
#include "layers_generic.h"
AI_API_DECLARE_BEGIN
/******************************************************************************/
/* Forward Functions Section */
/******************************************************************************/
/*!
* @brief Compute the indices of the max elements of the input tensor's element along the provided axis.
* @ingroup layers_generic
* @param layer argminmax layer
*/
AI_INTERNAL_API
void forward_argmax_is8(ai_layer* layer);
/*!
* @brief Compute the indices of the max elements of the input tensor's element along the provided axis.
* @ingroup layers_generic
* @param layer argminmax layer
*/
AI_INTERNAL_API
void forward_argmin_is8(ai_layer* layer);
AI_API_DECLARE_END
#endif /*LAYERS_ARGMINMAX_H*/