mirror of
https://github.com/openmv/openmv.git
synced 2025-11-04 14:49:50 +08:00
59 lines
2.6 KiB
C
59 lines
2.6 KiB
C
/**
|
|
******************************************************************************
|
|
* @file lite_upsample.h
|
|
* @author AIS
|
|
* @brief header file of AI platform lite upsample kernel datatypes
|
|
******************************************************************************
|
|
* @attention
|
|
*
|
|
* Copyright (c) 2021 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 LITE_UPSAMPLE_GENERIC_H
|
|
#define LITE_UPSAMPLE_GENERIC_H
|
|
|
|
|
|
#include "ai_lite_interface.h"
|
|
|
|
|
|
void forward_lite_upsample_generic_nearest(const ai_u8* in_data,
|
|
ai_u8* out_data,
|
|
const ai_size width_in,
|
|
const ai_size width_out,
|
|
const ai_float width_scale,
|
|
const ai_size height_out,
|
|
const ai_float height_scale,
|
|
const ai_u32 output_tensor_w_stride,
|
|
const ai_float offset_round_coeff);
|
|
|
|
void forward_lite_upsample_nearest(ai_ptr in_data,
|
|
ai_ptr out_data,
|
|
const ai_size width_in,
|
|
const ai_size height_in,
|
|
const ai_float width_scale,
|
|
const ai_float height_scale,
|
|
const ai_size width_out,
|
|
const ai_size height_out,
|
|
const ai_ptr_offset stride_w,
|
|
const ai_float offset_round_coeff);
|
|
|
|
void forward_lite_upsample_zeros( ai_ptr in_data,
|
|
ai_ptr out_data,
|
|
const ai_size width_in,
|
|
const ai_size height_in,
|
|
const ai_float width_scale,
|
|
const ai_float height_scale,
|
|
const ai_size width_out,
|
|
const ai_size height_out,
|
|
const ai_ptr_offset stride_ch,
|
|
const ai_ptr_offset stride_w,
|
|
const ai_handle p_zero_value);
|
|
|
|
#endif /*LITE_UPSAMPLE_GENERIC_H*/
|