diff options
author | Guo, Yejun <yejun.guo@intel.com> | 2021-01-07 11:14:10 +0800 |
---|---|---|
committer | Guo, Yejun <yejun.guo@intel.com> | 2021-01-15 08:59:54 +0800 |
commit | 64ea15f05075d58cbd76bdcd5bac9d3b47d891ee (patch) | |
tree | 9abe0da4c257d31f4e4ca85e0a9925f6acaaa966 /libavfilter/dnn_interface.h | |
parent | 57dae5723f2e06a531d7db8e961f84ed5d5f07de (diff) |
libavfilter/dnn: add batch mode for async execution
the default number of batch_size is 1
Signed-off-by: Xie, Lin <lin.xie@intel.com>
Signed-off-by: Wu Zhiwen <zhiwen.wu@intel.com>
Signed-off-by: Guo, Yejun <yejun.guo@intel.com>
Diffstat (limited to 'libavfilter/dnn_interface.h')
-rw-r--r-- | libavfilter/dnn_interface.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/libavfilter/dnn_interface.h b/libavfilter/dnn_interface.h index 9533c88829..ff338ea084 100644 --- a/libavfilter/dnn_interface.h +++ b/libavfilter/dnn_interface.h @@ -82,6 +82,8 @@ typedef struct DNNModule{ const char **output_names, uint32_t nb_output, AVFrame *out_frame); // Retrieve inference result. DNNAsyncStatusType (*get_async_result)(const DNNModel *model, AVFrame **in, AVFrame **out); + // Flush all the pending tasks. + DNNReturnType (*flush)(const DNNModel *model); // Frees memory allocated for model. void (*free_model)(DNNModel **model); } DNNModule; |