lavfi/dnn: Add nb_output to TaskItem

Add nb_output property to TaskItem for use in TensorFlow backend and Native backend.

Shubhanshu Saxena committed 5 years ago

Commit 9675ebb1 parent 446b4f72 files · +4 −0Browse files

Unified · Split

2 files

@@ -35,6 +35,7 @@ typedef struct TaskItem { · 34 unmodified lines
35	35	const char **output_names;
36	36	int async;
37	37	int do_ioproc;
	38	+ uint32_t nb_output;
38	39	uint32_t inference_todo;
39	40	uint32_t inference_done;
40	41	} TaskItem;

@@ -678,6 +678,7 @@ static DNNReturnType get_output_ov(void model, const char input_name, int inpu · 677 unmodified lines
678	678	task.in_frame = in_frame;
679	679	task.output_names = &output_name;
680	680	task.out_frame = out_frame;
	681	+ task.nb_output = 1;
681	682	task.model = ov_model;
682	683
683	684	if (extract_inference_from_task(ov_model->model->func_type, &task, ov_model->inference_queue, NULL) != DNN_SUCCESS) {
@@ -798,6 +799,7 @@ DNNReturnType ff_dnn_execute_model_ov(const DNNModel model, DNNExecBaseParams · 114 unmodified lines
798	799	task.in_frame = exec_params->in_frame;
799	800	task.output_names = &exec_params->output_names[0];
800	801	task.out_frame = exec_params->out_frame ? exec_params->out_frame : exec_params->in_frame;
	802	+ task.nb_output = exec_params->nb_output;
801	803	task.model = ov_model;
802	804
803	805	if (extract_inference_from_task(ov_model->model->func_type, &task, ov_model->inference_queue, exec_params) != DNN_SUCCESS) {
@@ -845,6 +847,7 @@ DNNReturnType ff_dnn_execute_model_async_ov(const DNNModel *model, DNNExecBasePa · 41 unmodified lines
845	847	task->in_frame = exec_params->in_frame;
846	848	task->output_names = &exec_params->output_names[0];
847	849	task->out_frame = exec_params->out_frame ? exec_params->out_frame : exec_params->in_frame;
	850	+ task->nb_output = exec_params->nb_output;
848	851	task->model = ov_model;
849	852	if (ff_queue_push_back(ov_model->task_queue, task) < 0) {
850	853	av_freep(&task);

Keyboard shortcuts