mirror of
https://github.com/openmv/openmv.git
synced 2025-11-04 14:49:50 +08:00
50 lines
1.5 KiB
C
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*/
|