40 lines
1.2 KiB
C
40 lines
1.2 KiB
C
/*
|
|
Copyright (c), 2001-2022, Shenshu Tech. Co., Ltd.
|
|
*/
|
|
|
|
#ifndef _SS_MPI_NNIE_H_
|
|
#define _SS_MPI_NNIE_H_
|
|
|
|
#include "ot_common_nnie.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
td_s32 ss_mpi_svp_nnie_load_model(const ot_svp_src_mem_info *model_buf, ot_svp_nnie_model *model);
|
|
|
|
td_s32 ss_mpi_svp_nnie_get_task_buf_size(td_u32 max_batch_num, td_u32 max_bbox_num,
|
|
const ot_svp_nnie_model *model, ot_svp_nnie_task_buf_info task_buf_info[], td_u32 net_seg_num);
|
|
|
|
td_s32 ss_mpi_svp_nnie_forward(ot_svp_nnie_handle *handle,
|
|
const ot_svp_src_blob src[], const ot_svp_nnie_model *model, const ot_svp_dst_blob dst[],
|
|
const ot_svp_nnie_forward_ctrl *forward_ctrl, td_bool is_instant);
|
|
|
|
td_s32 ss_mpi_svp_nnie_unload_model(ot_svp_nnie_model *model);
|
|
|
|
td_s32 ss_mpi_svp_nnie_set_prior_box(const ot_svp_src_blob prior_box[],
|
|
td_u32 prior_box_num, ot_svp_nnie_model *model);
|
|
|
|
td_s32 ss_mpi_svp_nnie_query(ot_svp_nnie_id nnie_id, ot_svp_nnie_handle handle,
|
|
td_bool *is_finish, td_bool is_block);
|
|
|
|
td_s32 ss_mpi_svp_nnie_add_mem_info(const ot_svp_mem_info *mem_info);
|
|
|
|
td_s32 ss_mpi_svp_nnie_rm_mem_info(const ot_svp_mem_info *mem_info);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif
|