11bd4fe43Sopenharmony_ci/* 21bd4fe43Sopenharmony_ci * Copyright (c) 2022 HiSilicon (Shanghai) Technologies CO., LIMITED. 31bd4fe43Sopenharmony_ci * Licensed under the Apache License, Version 2.0 (the "License"); 41bd4fe43Sopenharmony_ci * you may not use this file except in compliance with the License. 51bd4fe43Sopenharmony_ci * You may obtain a copy of the License at 61bd4fe43Sopenharmony_ci * 71bd4fe43Sopenharmony_ci * http://www.apache.org/licenses/LICENSE-2.0 81bd4fe43Sopenharmony_ci * 91bd4fe43Sopenharmony_ci * Unless required by applicable law or agreed to in writing, software 101bd4fe43Sopenharmony_ci * distributed under the License is distributed on an "AS IS" BASIS, 111bd4fe43Sopenharmony_ci * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 121bd4fe43Sopenharmony_ci * See the License for the specific language governing permissions and 131bd4fe43Sopenharmony_ci * limitations under the License. 141bd4fe43Sopenharmony_ci */ 151bd4fe43Sopenharmony_ci 161bd4fe43Sopenharmony_ci#ifndef _HI_MPI_NNIE_H_ 171bd4fe43Sopenharmony_ci#define _HI_MPI_NNIE_H_ 181bd4fe43Sopenharmony_ci 191bd4fe43Sopenharmony_ci#include "hi_nnie.h" 201bd4fe43Sopenharmony_ci 211bd4fe43Sopenharmony_ci#ifdef __cplusplus 221bd4fe43Sopenharmony_ci#if __cplusplus 231bd4fe43Sopenharmony_ciextern "C" { 241bd4fe43Sopenharmony_ci#endif 251bd4fe43Sopenharmony_ci#endif /* End of #ifdef __cplusplus */ 261bd4fe43Sopenharmony_ci 271bd4fe43Sopenharmony_ciHI_S32 HI_MPI_SVP_NNIE_LoadModel(const SVP_SRC_MEM_INFO_S *pstModelBuf, SVP_NNIE_MODEL_S *pstModel); 281bd4fe43Sopenharmony_ci 291bd4fe43Sopenharmony_ciHI_S32 HI_MPI_SVP_NNIE_GetTskBufSize(HI_U32 u32MaxInputNum, HI_U32 u32MaxBboxNum, 301bd4fe43Sopenharmony_ci const SVP_NNIE_MODEL_S *pstModel, HI_U32 au32TskBufSize[], HI_U32 u32NetSegNum); 311bd4fe43Sopenharmony_ci 321bd4fe43Sopenharmony_ciHI_S32 HI_MPI_SVP_NNIE_Forward(SVP_NNIE_HANDLE *phSvpNnieHandle, 331bd4fe43Sopenharmony_ci const SVP_SRC_BLOB_S astSrc[], const SVP_NNIE_MODEL_S *pstModel, const SVP_DST_BLOB_S astDst[], 341bd4fe43Sopenharmony_ci const SVP_NNIE_FORWARD_CTRL_S *pstForwardCtrl, HI_BOOL bInstant); 351bd4fe43Sopenharmony_ci 361bd4fe43Sopenharmony_ciHI_S32 HI_MPI_SVP_NNIE_ForwardWithBbox(SVP_NNIE_HANDLE *phSvpNnieHandle, 371bd4fe43Sopenharmony_ci const SVP_SRC_BLOB_S astSrc[], const SVP_SRC_BLOB_S astBbox[], const SVP_NNIE_MODEL_S *pstModel, 381bd4fe43Sopenharmony_ci const SVP_DST_BLOB_S astDst[], const SVP_NNIE_FORWARD_WITHBBOX_CTRL_S *pstForwardCtrl, HI_BOOL bInstant); 391bd4fe43Sopenharmony_ci 401bd4fe43Sopenharmony_ciHI_S32 HI_MPI_SVP_NNIE_UnloadModel(SVP_NNIE_MODEL_S *pstModel); 411bd4fe43Sopenharmony_ci 421bd4fe43Sopenharmony_ciHI_S32 HI_MPI_SVP_NNIE_Query(SVP_NNIE_ID_E enNnieId, SVP_NNIE_HANDLE svpNnieHandle, 431bd4fe43Sopenharmony_ci HI_BOOL *pbFinish, HI_BOOL bBlock); 441bd4fe43Sopenharmony_ci 451bd4fe43Sopenharmony_ciHI_S32 HI_MPI_SVP_NNIE_AddTskBuf(const SVP_MEM_INFO_S *pstTskBuf); 461bd4fe43Sopenharmony_ci 471bd4fe43Sopenharmony_ciHI_S32 HI_MPI_SVP_NNIE_RemoveTskBuf(const SVP_MEM_INFO_S *pstTskBuf); 481bd4fe43Sopenharmony_ci 491bd4fe43Sopenharmony_ci#ifdef __cplusplus 501bd4fe43Sopenharmony_ci#if __cplusplus 511bd4fe43Sopenharmony_ci} 521bd4fe43Sopenharmony_ci#endif 531bd4fe43Sopenharmony_ci#endif 541bd4fe43Sopenharmony_ci#endif /* _HI_MPI_NNIE_H_ */ 551bd4fe43Sopenharmony_ci 56